Drop deprecated classify

Author: billytrend-cohere

tests/test_async_client.py

@@ -29,7 +29,7 @@ async def test_chat(self) -> None:
                 UserMessage(
                     message="Who discovered gravity?"),
                 ChatbotMessage(message="The man who is widely credited with discovering "
-                                "gravity is Sir Isaac Newton")
+                               "gravity is Sir Isaac Newton")
             ],
             message="What year was he born?",
             connectors=[ChatConnector(id="web-search")]
@@ -43,7 +43,7 @@ async def test_chat_stream(self) -> None:
                 UserMessage(
                     message="Who discovered gravity?"),
                 ChatbotMessage(message="The man who is widely credited with discovering "
-                                "gravity is Sir Isaac Newton")
+                               "gravity is Sir Isaac Newton")
             ],
             message="What year was he born?",
             connectors=[ChatConnector(id="web-search")]
@@ -73,7 +73,8 @@ async def test_deprecated_fn(self) -> None:
 
     async def test_moved_fn(self) -> None:
         with self.assertRaises(ValueError):
-            await self.co.list_connectors("dummy", dummy="dummy")   # type: ignore
+            # type: ignore
+            await self.co.list_connectors("dummy", dummy="dummy")
 
     @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_generate(self) -> None:
@@ -176,35 +177,6 @@ async def test_rerank(self) -> None:
 
         print(response)
 
-    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
-    async def test_classify(self) -> None:
-        examples = [
-            ClassifyExample(
-                text="Dermatologists don't like her!", label="Spam"),
-            ClassifyExample(text="'Hello, open to this?'", label="Spam"),
-            ClassifyExample(
-                text="I need help please wire me $1000 right now", label="Spam"),
-            ClassifyExample(text="Nice to know you ;)", label="Spam"),
-            ClassifyExample(text="Please help me?", label="Spam"),
-            ClassifyExample(
-                text="Your parcel will be delivered today", label="Not spam"),
-            ClassifyExample(
-                text="Review changes to our Terms and Conditions", label="Not spam"),
-            ClassifyExample(text="Weekly sync notes", label="Not spam"),
-            ClassifyExample(
-                text="'Re: Follow up from today's meeting'", label="Not spam"),
-            ClassifyExample(text="Pre-read for tomorrow", label="Not spam"),
-        ]
-        inputs = [
-            "Confirm your email address",
-            "hey i need u to send some $",
-        ]
-        response = await self.co.classify(
-            inputs=inputs,
-            examples=examples,
-        )
-        print(response)
-
     @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_datasets_crud(self) -> None:
         my_dataset = await self.co.datasets.create(

tests/test_client.py

@@ -28,7 +28,7 @@ def test_chat(self) -> None:
                 UserMessage(
                     message="Who discovered gravity?"),
                 ChatbotMessage(message="The man who is widely credited with discovering "
-                                        "gravity is Sir Isaac Newton")
+                               "gravity is Sir Isaac Newton")
             ],
             message="What year was he born?",
             connectors=[ChatConnector(id="web-search")]
@@ -64,7 +64,7 @@ def test_chat_stream(self) -> None:
                 UserMessage(
                     message="Who discovered gravity?"),
                 ChatbotMessage(message="The man who is widely credited with discovering "
-                                        "gravity is Sir Isaac Newton")
+                               "gravity is Sir Isaac Newton")
             ],
             message="What year was he born?",
             connectors=[ChatConnector(id="web-search")]
@@ -238,35 +238,6 @@ def test_rerank(self) -> None:
 
         print(response)
 
-    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
-    def test_classify(self) -> None:
-        examples = [
-            ClassifyExample(
-                text="Dermatologists don't like her!", label="Spam"),
-            ClassifyExample(text="'Hello, open to this?'", label="Spam"),
-            ClassifyExample(
-                text="I need help please wire me $1000 right now", label="Spam"),
-            ClassifyExample(text="Nice to know you ;)", label="Spam"),
-            ClassifyExample(text="Please help me?", label="Spam"),
-            ClassifyExample(
-                text="Your parcel will be delivered today", label="Not spam"),
-            ClassifyExample(
-                text="Review changes to our Terms and Conditions", label="Not spam"),
-            ClassifyExample(text="Weekly sync notes", label="Not spam"),
-            ClassifyExample(
-                text="'Re: Follow up from today's meeting'", label="Not spam"),
-            ClassifyExample(text="Pre-read for tomorrow", label="Not spam"),
-        ]
-        inputs = [
-            "Confirm your email address",
-            "hey i need u to send some $",
-        ]
-        response = co.classify(
-            inputs=inputs,
-            examples=examples,
-        )
-        print(response)
-
     @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_datasets_crud(self) -> None:
         my_dataset = co.datasets.create(