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