From 428fc9a987888b6f22eec9eaca85ee0d386e902f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Fri, 1 Sep 2023 11:39:05 +0200
Subject: [PATCH] change reverted; proper fix sent upstream
---
pyproject.toml | 2 +-
src/maggotuba/models/predict_model.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index c5cfa6f..13ddec8 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 9aa35ad..086e4b8 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
--
GitLab