[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
0130c1d1ff
commit
b931b0039b
18 changed files with 107 additions and 74 deletions
|
|
@ -168,6 +168,6 @@ def test_text_and_image_vectors_share_dimension(monkeypatch):
|
|||
image_vectors = embeddings_module.encode_images([buf.getvalue()])
|
||||
text_vectors = embeddings_module.encode(["a blue square"])
|
||||
|
||||
assert image_vectors[0].shape == text_vectors[0].shape, (
|
||||
f"text dim {text_vectors[0].shape} != image dim {image_vectors[0].shape}"
|
||||
)
|
||||
assert (
|
||||
image_vectors[0].shape == text_vectors[0].shape
|
||||
), f"text dim {text_vectors[0].shape} != image dim {image_vectors[0].shape}"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue