[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Этот коммит содержится в:
pre-commit-ci[bot] 2023-02-28 15:21:34 +00:00
родитель 097142fd36
Коммит a6ea571478

Просмотреть файл

@ -22,27 +22,9 @@ def test_read_img():
test_img = ms.read_img(TEST_IMAGE_2)
assert numpy.array(test_img)[257][34] == [70, 66, 63]
#def test_load_feature_extractor_model_blip2():
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
# def test_load_feature_extractor_model_blip2():
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
mydict = {}
for img_path in images:
@ -54,4 +36,4 @@ def test_read_img():
keys = list(mydict.keys())
assert len(mydict) == 12
for key in keys:
assert len(mydict[key]["3_non-deterministic summary"]) == 3
assert len(mydict[key]["3_non-deterministic summary"]) == 3