bugfix in predict_model for hdf5 larva_dataset files

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:

src/maggotuba/models/predict_model.py

@@ -150,7 +150,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels):
 def predict_larva_dataset(backend, model, file, subset="validation", subsets=(.8, .2, 0)):
     from taggingbackends.data.dataset import LarvaDataset
     dataset = LarvaDataset(file, new_generator(), subsets)
-    return model.predict(dataset, subset)
+    return model.predict((dataset, subset))
 
 def _zip(xs, ys):
     # prevent issues similar to #2