From 7720020248a0c2fb2b08c2266fc3076d3384fda7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20=20LAURENT?= <francois.laurent@pasteur.fr>
Date: Fri, 1 Sep 2023 11:41:47 +0200
Subject: [PATCH] Direct calls to predict_model with hdf5 larva_dataset files
---
pyproject.toml | 4 ++--
src/maggotuba/models/modules.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index 2f85c89..13ddec8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
[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"]
diff --git a/src/maggotuba/models/modules.py b/src/maggotuba/models/modules.py
index 62ca070..f567e73 100644
--- a/src/maggotuba/models/modules.py
+++ b/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:
--
GitLab