From 38828ae7055f3a7723074e9a800bc367d885d866 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 29 Aug 2023 21:20:57 -0700
Subject: [PATCH] remove flaky petals test

---
 litellm/tests/test_completion.py | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 3b451a2f3a2..6ffef16a6cb 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -356,19 +356,19 @@ def test_completion_together_ai():
 #     except Exception as e:
 #         pytest.fail(f"Error occurred: {e}")
 
-def test_petals():
-    model_name = "stabilityai/StableBeluga2"
-    try:
-        response = completion(
-            model=model_name,
-            messages=messages,
-            custom_llm_provider="petals",
-            force_timeout=120,
-        )
-        # Add any assertions here to check the response
-        print(response)
-    except Exception as e:
-        pytest.fail(f"Error occurred: {e}")
+# def test_petals():
+#     model_name = "stabilityai/StableBeluga2"
+#     try:
+#         response = completion(
+#             model=model_name,
+#             messages=messages,
+#             custom_llm_provider="petals",
+#             force_timeout=120,
+#         )
+#         # Add any assertions here to check the response
+#         print(response)
+#     except Exception as e:
+#         pytest.fail(f"Error occurred: {e}")
 
 
 def test_completion_with_fallbacks():