CU-8693u6b4u: Make deid tests run with cpu-only torch

CogStack · Feb 13, 2024 · 58d4e93 · 58d4e93
1 parent 5927b25
commit 58d4e93
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -7,3 +7,4 @@ types-aiofiles==0.8.3
 types-PyYAML==6.0.3
 types-setuptools==57.4.10
 timeout-decorator==0.5.0
+instld~=0.0.24
diff --git a/tests/utils/ner/test_deid.py b/tests/utils/ner/test_deid.py
@@ -1,17 +1,23 @@
 from medcat.utils.ner import deid
-from medcat.utils.ner import make_or_update_cdb
+import instld
 
-from medcat.ner import transformers_ner
+from .helper import TORCH_VERSION, CPU_ONLY_TORCH_URL
 
-from spacy.tokens import Doc, Span
+with instld('torch==' + TORCH_VERSION, catch_output=True,
+            index_url=CPU_ONLY_TORCH_URL):
+    from medcat.utils.ner import make_or_update_cdb
 
-from typing import Any, List, Tuple
-import os
-import json
-import tempfile
+    from medcat.ner import transformers_ner
 
-import unittest
-import timeout_decorator
+    from spacy.tokens import Doc, Span
+
+    from typing import Any, List, Tuple
+    import os
+    import json
+    import tempfile
+
+    import unittest
+    import timeout_decorator
 
 FILE_DIR = os.path.dirname(os.path.realpath(__file__))