diff --git a/pyproject.toml b/pyproject.toml index 67c8cd6ea38d1c4eb6319270a538b5ba35b5eacf..f3efe95b12b03a967c841c392b8fce5ee66d973b 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 8595288f58dbe4a2619f5922237632488b902978..b193efc21a784515113a011205e803f297a3b7d7 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}")