phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) remove dup hf test

Browse source
This commit is contained in:
ishaan-jaff 2023-11-08 09:54:40 -08:00
parent 6d85d6486b
commit 739e9c4732
1 changed files with 0 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

21
litellm/tests/test_text_completion.py
View file

@ -123,24 +123,3 @@ def test_completion_hf_prompt_array():
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_hf_prompt_array() test_completion_hf_prompt_array()
def test_completion_hf_prompt_array():
try:
litellm.set_verbose=False
print("\n testing hf mistral\n")
response = text_completion(
model="huggingface/mistralai/Mistral-7B-v0.1",
prompt=token_prompt, # token prompt is a 2d list,
max_tokens=0,
temperature=0.0,
echo=True,
)
print("\n\n response")
print(response)
print(response.choices)
assert(len(response.choices)==2)
# response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_hf_prompt_array()