labelr 0.5.0__tar.gz → 0.6.0__tar.gz

{labelr-0.5.0/src/labelr.egg-info → labelr-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: labelr
-Version: 0.5.0
+Version: 0.6.0
 Summary: A command-line tool to manage labeling tasks with Label Studio.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{labelr-0.5.0 → labelr-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "labelr"
-version = "0.5.0"
+version = "0.6.0"
 description = "A command-line tool to manage labeling tasks with Label Studio."
 readme = "README.md"
 requires-python = ">=3.10"

{labelr-0.5.0 → labelr-0.6.0}/src/labelr/apps/train.py

@@ -128,7 +128,7 @@ def launch_job(
     # will run on.
     policy = batch_v1.AllocationPolicy.InstancePolicy()
     # See list of machine types here:
-    # https://docs.cloud.google.com/compute/docs/gpus#l4-gpus
+    # https://docs.cloud.google.com/compute/docs/gpus#t4-gpus
     policy.machine_type = machine_type
 
     accelerator = batch_v1.AllocationPolicy.Accelerator()

{labelr-0.5.0 → labelr-0.6.0}/src/labelr/export.py

@@ -240,13 +240,22 @@ def export_from_hf_to_ultralytics_object_detection(
     data_dir.mkdir(parents=True, exist_ok=True)
     category_id_to_name = {}
 
+    split_map = {
+        "train": "train",
+        "val": "val",
+    }
+    if "val" not in ds and "test" in ds:
+        logger.info("val split not found, using test split instead as val")
+        split_map["val"] = "test"
+
     for split in ["train", "val"]:
+        split_target = split_map[split]
         split_labels_dir = data_dir / "labels" / split
         split_labels_dir.mkdir(parents=True, exist_ok=True)
         split_images_dir = data_dir / "images" / split
         split_images_dir.mkdir(parents=True, exist_ok=True)
 
-        for sample in tqdm.tqdm(ds[split], desc="samples"):
+        for sample in tqdm.tqdm(ds[split_target], desc="samples"):
             image_id = sample["image_id"]
 
             if download_images:

{labelr-0.5.0 → labelr-0.6.0/src/labelr.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: labelr
-Version: 0.5.0
+Version: 0.6.0
 Summary: A command-line tool to manage labeling tasks with Label Studio.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown