qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-05-26 07:39:22 +02:00
Code Issues Projects Releases Wiki Activity

Revert "test_model_zoo: fix calls"

Browse source 
This reverts commit 5a98f55be3.
This commit is contained in:
Robert Sachunsky 2026-05-19 03:32:37 +02:00
parent a1449da1d1
commit 7e8b9311d3
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_model_zoo.py
View file

@ -6,11 +6,11 @@ def test_trocr1(
model_zoo = EynollahModelZoo(model_dir) model_zoo = EynollahModelZoo(model_dir)
try: try:
from transformers import TrOCRProcessor, VisionEncoderDecoderModel from transformers import TrOCRProcessor, VisionEncoderDecoderModel
model_zoo.load_models('trocr_processor') model_zoo.load_model('trocr_processor')
proc = model_zoo.get('trocr_processor') proc = model_zoo.get('trocr_processor', TrOCRProcessor)
assert isinstance(proc, TrOCRProcessor) assert isinstance(proc, TrOCRProcessor)
model_zoo.load_models(['ocr', 'tr']) model_zoo.load_model('ocr', 'tr')
model = model_zoo.get('ocr') model = model_zoo.get('ocr', VisionEncoderDecoderModel)
assert isinstance(model, VisionEncoderDecoderModel) assert isinstance(model, VisionEncoderDecoderModel)
except ImportError: except ImportError:
pass pass