test: fix test_logging typehint syntax error (#9142)

(cherry picked from commit 3f7a396)
determined-ai · Apr 10, 2024 · 691d190 · 691d190
1 parent 61999f4
commit 691d190
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/e2e_tests/tests/cluster/test_logging.py b/e2e_tests/tests/cluster/test_logging.py
@@ -18,7 +18,7 @@
 LogFields = Union[bindings.v1TaskLogsFieldsResponse, bindings.v1TrialLogsFieldsResponse]
 
 
-def _test_trial_logs(log_regex: re.Pattern[str]) -> None:
+def _test_trial_logs(log_regex: re.Pattern) -> None:
     sess = api_utils.user_session()
 
     experiment_id = exp.run_basic_test(
@@ -69,9 +69,7 @@ def test_trial_logs() -> None:
     _test_trial_logs(log_regex)
 
 
-def _test_task_logs(
-    task_type: str, task_config: Dict[str, Any], log_regex: re.Pattern[str]
-) -> None:
+def _test_task_logs(task_type: str, task_config: Dict[str, Any], log_regex: re.Pattern) -> None:
     sess = api_utils.user_session()
 
     rps = bindings.get_GetResourcePools(sess)
@@ -177,7 +175,7 @@ def test_task_logs(task_type: str, task_config: Dict[str, Any], log_regex: Any)
 
 
 def check_logs(
-    log_regex: Any,
+    log_regex: re.Pattern,
     log_fn: Callable[..., Iterable[Log]],
     log_fields_fn: Callable[..., Iterable[LogFields]],
 ) -> None: