use burn::{
module::Module,
prelude::Backend,
};
use crate::errors::{
BunsenResult,
WithOkOrPanic,
};
pub trait ModuleInit<B: Backend, M: Module<B>> {
fn try_init(
&self,
device: &B::Device,
) -> BunsenResult<M>;
fn init(
&self,
device: &B::Device,
) -> M {
self.try_init(device).ok_or_panic()
}
}