Skip to content

Commit f3a04b0

Browse files
authored
Install Timm package (#1207)
http://b/266100526
1 parent 102fb40 commit f3a04b0

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

Dockerfile.tmpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -524,6 +524,7 @@ RUN pip install flashtext && \
524524
pip install onnx && \
525525
pip install tables && \
526526
pip install openpyxl && \
527+
pip install timm && \
527528
/tmp/clean-layer.sh
528529

529530
# Download base easyocr models.

tests/test_timm.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
import unittest
2+
3+
import timm
4+
5+
6+
class TestTimm(unittest.TestCase):
7+
def test_list_models(self):
8+
models = timm.list_models()
9+
self.assertGreater(len(models), 0)

0 commit comments

Comments
 (0)