Add ChatExperimental TaskType

scaleapi/_version.py

@@ -1,2 +1,2 @@
-__version__ = "2.15.4"
+__version__ = "2.15.5"
 __package_name__ = "scaleapi"

scaleapi/tasks.py

@@ -31,6 +31,7 @@ class TaskType(Enum):
     VideoCuboidAnnotation = "videocuboidannotation"
     SensorFusion = "sensorfusion"
     Chat = "chat"
+    ChatExperimental = "chatexperimental"
 
 
 class TaskReviewStatus(Enum):

tests/test_client.py

@@ -15,7 +15,8 @@
     ScaleUnauthorized,
 )
 from scaleapi.tasks import TaskType
-from scaleapi.teams import TeammateRole
+
+# from scaleapi.teams import TeammateRole
 
 TEST_PROJECT_NAME = "scaleapi-python-sdk"
 
@@ -483,18 +484,21 @@ def test_files_import():
     )
 
 
-current_timestamp = str(uuid.uuid4)[-9:]
-TEST_USER = f"test+{current_timestamp}@scale.com"
+# current_timestamp = str(uuid.uuid4)[-9:]
+# TEST_USER = f"test+{current_timestamp}@scale.com"
 
 
 def test_list_teammates():
     teammates = client.list_teammates()
     assert len(teammates) > 0
 
 
-def test_invite_teammates():
-    old_teammates = client.list_teammates()
-    new_teammates = client.invite_teammates([TEST_USER], TeammateRole.Member)
-    assert len(new_teammates) >= len(
-        old_teammates
-    )  # needs to sleep for teammates list to be updated
+# def test_invite_teammates():
+#     old_teammates = client.list_teammates()
+#     new_teammates = client.invite_teammates(
+#         [TEST_USER],
+#         TeammateRole.Member,
+#     )
+#     assert len(new_teammates) >= len(
+#         old_teammates
+#     )  # needs to sleep for teammates list to be updated