From f9c26602fba62b00414c4920f68958b72e5d26d6 Mon Sep 17 00:00:00 2001 From: mart-r Date: Tue, 13 Feb 2024 14:23:49 +0000 Subject: [PATCH] CU-8693u6b4u: Add mising test-helper file --- tests/utils/ner/helper.py | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 tests/utils/ner/helper.py diff --git a/tests/utils/ner/helper.py b/tests/utils/ner/helper.py new file mode 100644 index 000000000..ff63e79a6 --- /dev/null +++ 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/"