Remove unused ARGS in test_runner_base
ARGS is only used for the python hint. It should not be defined in the
test_runner_base.
Tests: m atest && atest-dev --host atest_unittests
Bug: 277866911
Change-Id: I0752cfeb82798ced2c17633bd9ce4c259ae33be2
diff --git a/atest/bazel_mode.py b/atest/bazel_mode.py
index 9eda5d3..b36a2bf 100644
--- a/atest/bazel_mode.py
+++ b/atest/bazel_mode.py
@@ -1766,8 +1766,8 @@
def parse_args(
test_infos: List[test_info.TestInfo],
- extra_args: trb.ARGS,
- mod_info: module_info.ModuleInfo) -> trb.ARGS:
+ extra_args: Dict[str, Any],
+ mod_info: module_info.ModuleInfo) -> Dict[str, Any]:
"""Parse commandline args and passes supported args to bazel.
Args:
diff --git a/atest/test_runners/atest_tf_test_runner.py b/atest/test_runners/atest_tf_test_runner.py
index cb10e40..499c81c 100644
--- a/atest/test_runners/atest_tf_test_runner.py
+++ b/atest/test_runners/atest_tf_test_runner.py
@@ -29,7 +29,7 @@
from functools import partial
from pathlib import Path
-from typing import Any, List, Set, Tuple
+from typing import Any, Dict, List, Set, Tuple
from atest import atest_configs
from atest import atest_error
@@ -1033,9 +1033,11 @@
return annotation_filter_args
-def extra_args_to_tf_args(mod_info: module_info.ModuleInfo,
- test_infos: List[test_info.TestInfo],
- extra_args: trb.ARGS) -> Tuple[trb.ARGS, trb.ARGS]:
+def extra_args_to_tf_args(
+ mod_info: module_info.ModuleInfo,
+ test_infos: List[test_info.TestInfo],
+ extra_args: Dict[str, Any],
+) -> Tuple[Dict[str, Any], Dict[str, Any]]:
"""Convert the extra args into atest_tf_test_runner supported args.
Args:
diff --git a/atest/test_runners/test_runner_base.py b/atest/test_runners/test_runner_base.py
index 177c791..cfddfe0 100644
--- a/atest/test_runners/test_runner_base.py
+++ b/atest/test_runners/test_runner_base.py
@@ -49,8 +49,6 @@
IGNORED_STATUS = 'IGNORED'
ERROR_STATUS = 'ERROR'
-ARGS = Dict[str, Any]
-
class TestRunnerBase:
"""Base Test Runner class."""
