phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix triton embedding test

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-17 17:29:22 -07:00
parent 03f79f78db
commit 613bbe306f
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_embedding.py
View file

@ -589,7 +589,7 @@ async def test_triton_embeddings():
print(f"response: {response}") print(f"response: {response}")
# stubbed endpoint is setup to return this # stubbed endpoint is setup to return this
assert response.data[0]["embedding"] == [0.1, 0.2, 0.3] assert response.data[0]["embedding"] == [0.1, 0.2]
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")