From 08132975f4eebc6a38b27fdeed2e677d66706fc5 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 11 Sep 2023 08:44:23 -0700
Subject: [PATCH] update test_batch_completions

---
 litellm/tests/test_batch_completions.py | 49 ++++++++++++-------------
 1 file changed, 23 insertions(+), 26 deletions(-)

diff --git a/litellm/tests/test_batch_completions.py b/litellm/tests/test_batch_completions.py
index e53c015f3b..560a4a3d92 100644
--- a/litellm/tests/test_batch_completions.py
+++ b/litellm/tests/test_batch_completions.py
@@ -1,28 +1,25 @@
-#### What this tests ####
-#    This tests calling batch_completions by running 100 messages together
+# #### What this tests ####
+# #    This tests calling batch_completions by running 100 messages together
 
-import sys, os
-import traceback
-import pytest
-sys.path.insert(
-    0, os.path.abspath("../..")
-)  # Adds the parent directory to the system path
-from openai.error import Timeout
-import litellm
-from litellm import batch_completion
-litellm.set_verbose=True
+# import sys, os
+# import traceback
+# import pytest
+# sys.path.insert(
+#     0, os.path.abspath("../..")
+# )  # Adds the parent directory to the system path
+# from openai.error import Timeout
+# import litellm
+# from litellm import batch_completion
+# litellm.set_verbose=True
 
-def test_batch_completions():
-    messages = [[{"role": "user", "content": "Hey, how's it going"}] for _ in range(5)]
-    print(messages[0:5])
-    print(len(messages))
-    # model = "vllm/facebook/opt-125m"
-    model = "gpt-3.5-turbo"
-    try:
-        result = batch_completion(model=model, messages=messages)
-        print(result)
-        print(len(result))
-    except Timeout as e:
-        pass
-    except Exception as e:
-        pytest.fail(f"An error occurred: {e}")
\ No newline at end of file
+# def test_batch_completions():
+#     messages = [[{"role": "user", "content": "Hey, how's it going"}] for _ in range(5)]
+#     model = "gpt-3.5-turbo"
+#     try:
+#         result = batch_completion(model=model, messages=messages)
+#         print(result)
+#         print(len(result))
+#     except Timeout as e:
+#         pass
+#     except Exception as e:
+#         pytest.fail(f"An error occurred: {e}")
\ No newline at end of file