diff --git a/pyproject.toml b/pyproject.toml index c5cfa6f6f726e822f454d68c8044fec054513d0f..13ddec87d5983a97e9a3e8da7970d9d8dfda0215 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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.1"} +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/predict_model.py b/src/maggotuba/models/predict_model.py index 9aa35ad8a127c3b88fb70441e224469e2baa3249..086e4b8d2a4be887c5aa6791fa0325268fe649ae 100644 --- a/src/maggotuba/models/predict_model.py +++ b/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