Direct calls to predict_model with hdf5 larva_dataset files

pyproject.toml

 [tool.poetry]
 name = "MaggotUBA-adapter"
-version = "0.16.2"
+version = "0.16.3"
 description = "Interface between MaggotUBA and the Nyx tagging UI"
 authors = ["François Laurent"]
 license = "MIT"
@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v
 torch = "^1.11.0"
 numpy = "^1.19.3"
 protobuf = "3.9.2"
-taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15"}
+taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15.2"}
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

src/maggotuba/models/modules.py

@@ -362,7 +362,7 @@ class MaggotClassifier(MaggotModule):
             if n_iterations is not None:
                 if isinstance(n_iterations, str):
                     n_iterations = map(int, n_iterations.split(','))
-                if isinstance(n_iterations, int):
+                if isinstance(n_iterations, (int, np.int64)):
                     n_pretraining_iter = n_iterations // 2
                     n_finetuning_iter = n_iterations // 2
                 else: