qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-06-16 09:59:13 +02:00
Code Issues Projects Releases Wiki Activity

models w/ multiple inputs yield a tuple for .input_shape

Browse source
This commit is contained in:
Robert Sachunsky 2026-06-12 14:55:46 +02:00
parent 08946067ac
commit 45c92eada2
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_model_zoo.py
View file

@ -23,6 +23,7 @@ def test_cnnrnnocr1(
model = model_zoo.get('ocr')
assert isinstance(model, Predictor)
shape = model.input_shape
assert len(shape) == 4
assert len(shape) == 2
assert len(shape[0]) == 4
except ImportError:
pass