testing github package

app.py

@@ -9,6 +9,8 @@ import matplotlib.pyplot as plt
 import numpy as np
 import optax
 
+from dcmnet.data import prepare_batches
+
 # Disable future warnings.
 import warnings
 warnings.simplefilter(action='ignore', category=FutureWarning)
@@ -21,7 +23,7 @@ rotation = e3x.so3.random_rotation(rotation_key)
 
 st.write(rotation)
 
-
+st.write(prepare_batches)
 
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)