for use as modules

src/maggotuba/models/predict_model.py

@@ -3,7 +3,7 @@ from taggingbackends.data.chore import load_spine
 import taggingbackends.data.fimtrack as fimtrack
 from taggingbackends.data.labels import Labels
 from taggingbackends.features.skeleton import get_5point_spines
-from denselayer import DenseLayer
+from maggotuba.models.denselayer import DenseLayer
 import numpy as np
 import json
 

src/maggotuba/models/train_model.py

 from taggingbackends.data.labels import Labels
 from taggingbackends.data.dataset import LarvaDataset
-from denselayer import DenseLayer, device
+from maggotuba.models.denselayer import DenseLayer, device
 import numpy as np
 import json
 import torch