From 0c895b76fe05eb5a85e8f9006ff4e172b985d04c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 21 Sep 2023 11:29:41 -0700 Subject: [PATCH] fix tests --- litellm/tests/test_ollama_local.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_ollama_local.py b/litellm/tests/test_ollama_local.py index 27beb81a4..f00f1a0d9 100644 --- a/litellm/tests/test_ollama_local.py +++ b/litellm/tests/test_ollama_local.py @@ -27,7 +27,7 @@ # except Exception as e: # pytest.fail(f"Error occurred: {e}") -# test_completion_ollama() +# # test_completion_ollama() # def test_completion_ollama_stream(): # user_message = "what is litellm?" @@ -47,7 +47,7 @@ # except Exception as e: # pytest.fail(f"Error occurred: {e}") -# test_completion_ollama_stream() +# # test_completion_ollama_stream() # async def test_completion_ollama_async_stream(): @@ -68,10 +68,8 @@ # except Exception as e: # pytest.fail(f"Error occurred: {e}") -# import asyncio -# asyncio.run(test_completion_ollama_async_stream()) - - +# # import asyncio +# # asyncio.run(test_completion_ollama_async_stream()) # def prepare_messages_for_chat(text: str) -> list: # messages = [