From 4a91424ed4b89a2a67bc8179cb818ed7abac91ce Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 15 Aug 2023 13:02:01 -0700
Subject: [PATCH] fix failing tests

---
 litellm/tests/test_completion.py | 1 +
 pyproject.toml                   | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 6cd1e83053..e8f3cea423 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -231,6 +231,7 @@ def test_baseten_falcon_7bcompletion_withbase():
         print(response)
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
+    litellm.api_base = None
 
 
 
diff --git a/pyproject.toml b/pyproject.toml
index 247d76ec64..240c55d6d0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.396"
+version = "0.1.397"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"