From b895a3c001324676da9f6fb1cb868bc1a55ca326 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Mon, 1 Aug 2022 10:01:13 +0200
Subject: [PATCH] for use as modules

---
 src/maggotuba/models/predict_model.py | 2 +-
 src/maggotuba/models/train_model.py   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/maggotuba/models/predict_model.py b/src/maggotuba/models/predict_model.py
index afcae80..9ed347f 100644
--- a/src/maggotuba/models/predict_model.py
+++ b/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
 
diff --git a/src/maggotuba/models/train_model.py b/src/maggotuba/models/train_model.py
index 5527b97..34c87f6 100644
--- a/src/maggotuba/models/train_model.py
+++ b/src/maggotuba/models/train_model.py
@@ -1,6 +1,6 @@
 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
-- 
GitLab