CU-8693u6b4u: Add mising test-helper file

CogStack · Feb 13, 2024 · f9c2660 · f9c2660
1 parent 58d4e93
commit f9c2660
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/utils/ner/helper.py b/tests/utils/ner/helper.py
@@ -0,0 +1,4 @@
+import pkg_resources
+
+TORCH_VERSION = pkg_resources.get_distribution("torch").version
+CPU_ONLY_TORCH_URL = "https://download.pytorch.org/whl/cpu/"