SNOW-1445536: Remove StringMethods and DatetimeProperties (#1670)

snowflakedb · May 29, 2024 · dbb0713 · dbb0713
1 parent a8208d9
commit dbb0713
Show file tree

Hide file tree

Showing 4 changed files with 255 additions and 605 deletions.
diff --git a/src/snowflake/snowpark/modin/pandas/series.py b/src/snowflake/snowpark/modin/pandas/series.py
@@ -2335,7 +2335,13 @@ def dt(self):  # noqa: RT01, D200
         if not is_datetime64_any_dtype(current_dtype):
             raise AttributeError("Can only use .dt accessor with datetimelike values")
 
-        from .series_utils import DatetimeProperties
+        from modin.pandas.series_utils import DatetimeProperties
+
+        if DatetimeProperties._Series is not Series:
+            del (
+                DatetimeProperties._Series
+            )  # Replace modin's Series class with Snowpark pandas Series
+            DatetimeProperties._Series = Series
 
         return DatetimeProperties(self)
 
@@ -2468,7 +2474,13 @@ def str(self):  # noqa: RT01, D200
         if not is_string_dtype(current_dtype):
             raise AttributeError("Can only use .str accessor with string values!")
 
-        from .series_utils import StringMethods
+        from modin.pandas.series_utils import StringMethods
+
+        if StringMethods._Series is not Series:
+            del (
+                StringMethods._Series
+            )  # Replace modin's Series class with Snowpark pandas Series
+            StringMethods._Series = Series
 
         return StringMethods(self)