Skip to content

Commit

Permalink
Merge branch 'ensure_return_type' of https://github.com/KennethEnevol…
Browse files Browse the repository at this point in the history
  • Loading branch information
KennethEnevoldsen committed Feb 17, 2024
2 parents e8d3994 + 06c5cd8 commit 7d09487
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_all_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def test_embedding_match_what_is_stated(model: seb.SebModel):
This test checks if the embedding size matches what is stated in the ModelMeta
"""
output = model.encoder.encode(["test"])
output_embedding_size = output.shape[1]
output_embedding_size = output.shape[1]
assert output_embedding_size == model.meta.embedding_size


Expand Down

0 comments on commit 7d09487

Please sign in to comment.