Merge branch 'dev' into 'main'

Set of commits to be tagged v0.16.1

See merge request !3

pyproject.toml

 [tool.poetry]
 name = "MaggotUBA-adapter"
-version = "0.16"
+version = "0.16.1"
 description = "Interface between MaggotUBA and the Nyx tagging UI"
 authors = ["François Laurent"]
 license = "MIT"
@@ -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"]

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}")