Skip to content

Commit

Permalink
TEST-#7419: Fix mypy
Browse files Browse the repository at this point in the history
Signed-off-by: sfc-gh-mvashishtha <[email protected]>
  • Loading branch information
sfc-gh-mvashishtha committed Jan 11, 2025
1 parent 1c4d173 commit 8b248f6
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions modin/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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
Expand Down

0 comments on commit 8b248f6

Please sign in to comment.