[QQC-506] Rename CATALOG to BATCHES in QueueMode

Author: mkozik-lb

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+# Version 0.0.0 (YYYY-MM-DD) - In Progress
+
+### Changed
+* Update QueueMode enum to support new value for QueueMode.Batch = `BATCHES`.
+
 # Version 3.27.2 (2022-10-04)
 
 ### Added

labelbox/schema/queue_mode.py

@@ -1,6 +1,16 @@
 from enum import Enum
 
+import logging
+
+logger = logging.getLogger(__name__)
+
 
 class QueueMode(str, Enum):
-    Batch = "CATALOG"
+    Batch = "BATCHES"
     Dataset = "DATA_SET"
+
+    @classmethod
+    def _missing_(cls, value):
+        # Parses the deprecated "CATALOG" value back to QueueMode.Batch.
+        if value == "CATALOG":
+            return QueueMode.Batch

tests/unit/test_queue_mode.py

@@ -0,0 +1,20 @@
+import pytest
+
+from labelbox.schema.queue_mode import QueueMode
+
+
+def test_parse_deprecated_catalog():
+    assert QueueMode("CATALOG") == QueueMode.Batch
+
+
+def test_parse_batches():
+    assert QueueMode("BATCHES") == QueueMode.Batch
+
+
+def test_parse_data_set():
+    assert QueueMode("DATA_SET") == QueueMode.Dataset
+
+
+def test_fails_for_unknown():
+    with pytest.raises(ValueError):
+        QueueMode("foo")