Skip to content

Commit

Permalink
CU-8694gzbn3: Remove timeouts from DEID tests in case they're the one…
Browse files Browse the repository at this point in the history
…s creating issues
  • Loading branch information
mart-r committed May 14, 2024
1 parent 9b02925 commit faaf7fb
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions tests/utils/ner/test_deid.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
import tempfile

import unittest
import timeout_decorator

FILE_DIR = os.path.dirname(os.path.realpath(__file__))

Expand Down Expand Up @@ -176,15 +175,13 @@ def assertTextHasBeenDeIded(self, text: str, redacted: bool):
return
raise AssertionError("None of the CUIs found")

@timeout_decorator.timeout(3 * 60) # 3 minutes max
def test_model_can_multiprocess_no_redact(self):
processed = self.deid_model.deid_multi_texts(self.data, n_process=self.processes)
self.assertEqual(len(processed), 5)
for tid, new_text in enumerate(processed):
with self.subTest(str(tid)):
self.assertTextHasBeenDeIded(new_text, redacted=False)

@timeout_decorator.timeout(3 * 60) # 3 minutes max
def test_model_can_multiprocess_redact(self):
processed = self.deid_model.deid_multi_texts(self.data, n_process=self.processes, redact=True)
self.assertEqual(len(processed), 5)
Expand Down

0 comments on commit faaf7fb

Please sign in to comment.