From 8b248f6ef0d552b9e997b88ffe1a62f77ba11189 Mon Sep 17 00:00:00 2001 From: sfc-gh-mvashishtha Date: Fri, 10 Jan 2025 17:46:55 -0800 Subject: [PATCH] TEST-#7419: Fix mypy Signed-off-by: sfc-gh-mvashishtha --- modin/__init__.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/modin/__init__.py b/modin/__init__.py index 50c20327694..4e1d99cd3c1 100644 --- a/modin/__init__.py +++ b/modin/__init__.py @@ -12,10 +12,7 @@ # governing permissions and limitations under the License. import warnings -from typing import TYPE_CHECKING, Any, Optional, Tuple, Type, Union - -if TYPE_CHECKING: - from .config import Engine, StorageFormat +from typing import Any, Optional, Tuple, Type, Union from . import _version @@ -37,9 +34,7 @@ def custom_formatwarning( warnings.filterwarnings("ignore", message="Large object of size") -def set_execution( - engine: Any = None, storage_format: Any = None -) -> Tuple["Engine", "StorageFormat"]: +def set_execution(engine: Any = None, storage_format: Any = None) -> Tuple[Any, Any]: """ Method to set the _pair_ of execution engine and storage format format simultaneously. This is needed because there might be cases where switching one by one would be