Skip to content

Commit

Permalink
SNOW-1628820 remove timedelta warning message
Browse files Browse the repository at this point in the history
  • Loading branch information
sfc-gh-azhan committed Aug 28, 2024
1 parent e83e700 commit 3da5e48
Show file tree
Hide file tree
Showing 6 changed files with 1,380 additions and 359 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,8 @@
import pandas as native_pd

from snowflake.snowpark.column import Column
from snowflake.snowpark.modin.plugin.utils.warning_message import WarningMessage
from snowflake.snowpark.types import DataType, LongType

TIMEDELTA_WARNING_MESSAGE = (
"Snowpark pandas support for Timedelta is not currently available."
)

"""Map Python type to its from_pandas method"""
_python_type_to_from_pandas: dict[type, Callable[[Any], Any]] = {}

Expand Down Expand Up @@ -128,9 +123,6 @@ class TimedeltaType(SnowparkPandasType, LongType):
)

def __init__(self) -> None:
# TODO(SNOW-1620452): Remove this warning message before releasing
# Timedelta support.
WarningMessage.single_warning(TIMEDELTA_WARNING_MESSAGE)
super().__init__()

@staticmethod
Expand Down
206 changes: 190 additions & 16 deletions tests/notebooks/modin/MIMICHealthcareDemo.ipynb

Large diffs are not rendered by default.

Loading

0 comments on commit 3da5e48

Please sign in to comment.