Skip to content

Commit

Permalink
fix: tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ZiniuYu committed Apr 12, 2023
1 parent 5b9b90c commit 66bbad9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,15 +87,15 @@ def test_split_img_txt_da(inputs):
[
Document(
uri='https://clip-as-service.jina.ai/_static/favicon.png',
).load_uri_to_blob(),
).load_uri_to_image_tensor(),
]
)
],
)
def test_preproc_image(inputs):
from clip_server.model import clip

preprocess_fn = clip._transform_ndarray(224)
preprocess_fn = clip._transform_blob(224)
da, pixel_values = preproc_image(inputs, preprocess_fn, drop_image_content=True)
assert len(da) == 1
assert not da[0].blob
Expand Down

0 comments on commit 66bbad9

Please sign in to comment.