fix failing tests

2023-08-15 13:02:01 -07:00 · 2023-08-15 13:02:01 -07:00 · 4a91424ed4
commit 4a91424ed4
parent b18b12afb1
2 changed files with 2 additions and 1 deletions
--- 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



--- 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"