arkindex-base-worker 0.3.7rc6__py3-none-any.whl → 0.3.7rc8__py3-none-any.whl

arkindex_worker/worker/metadata.py

@@ -108,17 +108,17 @@ class MetaDataMixin:
         return metadata["id"]
 
     @unsupported_cache
-    def create_metadatas(
+    def create_metadata_bulk(
         self,
         element: Element | CachedElement,
-        metadatas: list[dict[str, MetaType | str | int | float | None]],
+        metadata_list: list[dict[str, MetaType | str | int | float | None]],
     ) -> list[dict[str, str]]:
         """
         Create multiple metadata on an existing element.
         This method does not support cache.
 
         :param element: The element to create multiple metadata on.
-        :param metadatas: The list of dict whose keys are the following:
+        :param metadata_list: The list of dict whose keys are the following:
             - type: MetaType
             - name: str
             - value: str | int | float
@@ -128,13 +128,13 @@ class MetaDataMixin:
             element, Element | CachedElement
         ), "element shouldn't be null and should be of type Element or CachedElement"
 
-        assert metadatas and isinstance(
-            metadatas, list
-        ), "type shouldn't be null and should be of type list of Dict"
+        assert metadata_list and isinstance(
+            metadata_list, list
+        ), "metadata_list shouldn't be null and should be of type list of dict"
 
         # Make a copy to avoid modifying the metadata_list argument
         metas = []
-        for index, metadata in enumerate(metadatas):
+        for index, metadata in enumerate(metadata_list):
             assert isinstance(
                 metadata, dict
             ), f"Element at index {index} in metadata_list: Should be of type dict"

hooks/pre_gen_project.py

@@ -0,0 +1,3 @@
+# Normalize the slug to generate __package and __module private variables
+{{cookiecutter.update({"__package": cookiecutter.slug.lower().replace("_", "-")})}}  # noqa: F821
+{{cookiecutter.update({"__module": cookiecutter.slug.lower().replace("-", "_")})}}  # noqa: F821

tests/conftest.py

@@ -22,7 +22,7 @@ from arkindex_worker.cache import (
     create_version_table,
     init_cache_db,
 )
-from arkindex_worker.models import Artifact, Dataset
+from arkindex_worker.models import Artifact, Dataset, Set
 from arkindex_worker.worker import BaseWorker, DatasetWorker, ElementsWorker
 from arkindex_worker.worker.dataset import DatasetState
 from arkindex_worker.worker.transcription import TextOrientation
@@ -597,7 +597,7 @@ def mock_databases(tmp_path):
 @pytest.fixture()
 def default_dataset():
     return Dataset(
-        **{
+        {
             "id": "dataset_id",
             "name": "My dataset",
             "description": "A super dataset built by me",
@@ -608,11 +608,15 @@ def default_dataset():
             "task_id": "11111111-1111-1111-1111-111111111111",
             "created": "2000-01-01T00:00:00Z",
             "updated": "2000-01-01T00:00:00Z",
-        },
-        selected_sets=["set_1", "set_2", "set_3"],
+        }
     )
 
 
+@pytest.fixture()
+def default_train_set(default_dataset):
+    return Set(name="train", dataset=default_dataset)
+
+
 @pytest.fixture()
 def mock_dataset_worker(monkeypatch, mocker, _mock_worker_run_api):
     monkeypatch.setenv("PONOS_TASK", "my_task")
@@ -635,9 +639,10 @@ def mock_dev_dataset_worker(mocker):
         [
             "worker",
             "--dev",
-            "--dataset",
-            "11111111-1111-1111-1111-111111111111",
-            "22222222-2222-2222-2222-222222222222",
+            "--set",
+            "11111111-1111-1111-1111-111111111111:train",
+            "11111111-1111-1111-1111-111111111111:val",
+            "22222222-2222-2222-2222-222222222222:my_set",
         ],
     )