version increment

pyproject.toml

 [tool.poetry]
 name = "MaggotUBA-adapter"
-version = "0.2.0"
+version = "0.3.0"
 description = "Interface between MaggotUBA and the Nyx tagging UI"
 authors = ["François Laurent"]
 license = "MIT"

src/maggotuba/models/predict_model.py

 from taggingbackends.data.trxmat import TrxMat
 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 randomforest import RandomForest
@@ -46,13 +47,17 @@ def predict_model(backend):
             if run == "spine":
                 run, data = next(iter(data.items()))
             t = t[run]
+        elif file.name.endswith(".csv"):
+            print("assuming 30 fps")
+            t, data = fimtrack.read_spines(file, fps=30)
+            run = "NA"
         else:
             # TODO: support more file formats
             continue
         # downsample the skeleton
         if isinstance(data, dict):
             for larva in data:
-                data[larva] = np.vstack([get_5point_spines(spine) for spine in data[larva]])
+                data[larva] = get_5point_spines(data[larva])
         else:
             data = get_5point_spines(data)
         # load the model