From 21db7e8c94c96d2096e96905b838311b9d198e1b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Fri, 7 Jul 2023 01:40:06 +0200
Subject: [PATCH] balancing_strategy management moved to taggingbackends

---
 pyproject.toml                     | 2 +-
 src/maggotuba/data/make_dataset.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 67c8cd6..f3efe95 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.14.1"}
+taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15"}
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/src/maggotuba/data/make_dataset.py b/src/maggotuba/data/make_dataset.py
index 8595288..b193efc 100644
--- a/src/maggotuba/data/make_dataset.py
+++ b/src/maggotuba/data/make_dataset.py
@@ -59,10 +59,10 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False,
 
             print("generating a larva_dataset file...")
             # generate a larva_dataset_*.hdf5 file in data/interim/{instance}/
-            balance = isinstance(balancing_strategy, str) and balancing_strategy.lower() == 'maggotuba'
             out = backend.generate_dataset(backend.raw_data_dir(),
                                            past_future_extensions=False,
-                                           balance=balance, **kwargs)
+                                           balancing_strategy=balancing_strategy,
+                                           **kwargs)
             print(f"larva_dataset file generated: {out}")
 
 
-- 
GitLab