Restore test

Kiln-AI · Nov 25, 2024 · 66c0e63 · 66c0e63
1 parent 8c81889
commit 66c0e63
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/libs/core/kiln_ai/adapters/fine_tune/test_openai_finetune.py b/libs/core/kiln_ai/adapters/fine_tune/test_openai_finetune.py
@@ -57,11 +57,13 @@ def test_setup(openai_finetune):
             FineTuneStatusType.unknown,
             "Server connection error",
         ),
-        # (
-        #    openai.RateLimitError(message="Rate limit exceeded", body={}),
-        #    FineTuneStatusType.unknown,
-        #    "Rate limit exceeded",
-        # ),
+        (
+            openai.RateLimitError(
+                message="Rate limit exceeded", body={}, response=MagicMock()
+            ),
+            FineTuneStatusType.unknown,
+            "Rate limit exceeded",
+        ),
         (
             openai.APIStatusError(
                 "Not found",