From d6cb2beee2fab5b4f56c5fcc38b2358537304564 Mon Sep 17 00:00:00 2001
From: Nitya Kasturi <nitya.kasturi@scale.com>
Date: Tue, 21 May 2024 17:38:26 -0400
Subject: [PATCH] Add ChatExperimental TaskType (#85)

* Add support for the ChatExperimental task type
---
 scaleapi/_version.py |  2 +-
 scaleapi/tasks.py    |  1 +
 tests/test_client.py | 22 +++++++++++++---------
 3 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/scaleapi/_version.py b/scaleapi/_version.py
index 5c5f6a7..a7a9859 100644
--- a/scaleapi/_version.py
+++ b/scaleapi/_version.py
@@ -1,2 +1,2 @@
-__version__ = "2.15.4"
+__version__ = "2.15.5"
 __package_name__ = "scaleapi"
diff --git a/scaleapi/tasks.py b/scaleapi/tasks.py
index 5ff0194..f5e0847 100644
--- a/scaleapi/tasks.py
+++ b/scaleapi/tasks.py
@@ -31,6 +31,7 @@ class TaskType(Enum):
     VideoCuboidAnnotation = "videocuboidannotation"
     SensorFusion = "sensorfusion"
     Chat = "chat"
+    ChatExperimental = "chatexperimental"
 
 
 class TaskReviewStatus(Enum):
diff --git a/tests/test_client.py b/tests/test_client.py
index feddcd9..fd6d65b 100644
--- a/tests/test_client.py
+++ b/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,8 +484,8 @@ 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():
@@ -492,9 +493,12 @@ def test_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