chore: 🚀 Add new functions to __init__

src/leibnetz/nets/__init__.py

@@ -1,5 +1,12 @@
 from .attentive_scalenet import build_attentive_scale_net
 from .scalenet import build_scale_net
 from .unet import build_unet
+from .bio import (
+    convert_to_bio,
+    convert_to_backprop,
+    HebbsRule,
+    KrotovsRule,
+    OjasRule,
+)
 
 # from .resnet import build_resnet

src/leibnetz/nets/bio.py

@@ -245,11 +245,3 @@ def convert_to_backprop(model: LeibNet):
 
 
 # %%
-from leibnetz.nets import build_unet as leibnetz_unet
-
-model = convert_to_bio(leibnetz_unet(), KrotovsRule(), learning_rate=0.1)
-inputs = model.get_example_inputs(device="cuda")
-model = model.to("cuda")
-# %%
-outputs = model(inputs)
-# %%