Merge pull request #626 from Labelbox/ms/unassign-split

support unassigned splits

Author: msokoloff1

labelbox/schema/model_run.py

@@ -203,13 +203,7 @@ def assign_data_rows_to_split(self,
                                   timeout_seconds=120):
 
         split_value = split.value if isinstance(split, DataSplit) else split
-
-        if split_value == DataSplit.UNASSIGNED.value:
-            raise ValueError(
-                f"Cannot assign split value of `{DataSplit.UNASSIGNED.value}`.")
-
-        valid_splits = filter(lambda name: name != DataSplit.UNASSIGNED.value,
-                              DataSplit._member_names_)
+        valid_splits = DataSplit._member_names_
 
         if split_value not in valid_splits:
             raise ValueError(

tests/integration/annotation_import/test_model_run.py

@@ -146,13 +146,7 @@ def test_model_run_split_assignment(model_run, dataset, image_url):
     with pytest.raises(ValueError):
         model_run.assign_data_rows_to_split(data_row_ids, "INVALID SPLIT")
 
-    with pytest.raises(ValueError):
-        model_run.assign_data_rows_to_split(data_row_ids, DataSplit.UNASSIGNED)
-
-    for split in ["TRAINING", "TEST", "VALIDATION", *DataSplit]:
-        if split == DataSplit.UNASSIGNED:
-            continue
-
+    for split in ["TRAINING", "TEST", "VALIDATION", "UNASSIGNED", *DataSplit]:
         model_run.assign_data_rows_to_split(data_row_ids, split)
         counts = Counter()
         for data_row in model_run.model_run_data_rows():