cleanup

alteryx · Feb 15, 2024 · eecda15 · eecda15
1 parent ef54890
commit eecda15
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/featuretools/entityset/entityset.py b/featuretools/entityset/entityset.py
@@ -1218,8 +1218,6 @@ def add_last_time_indexes(self, updated_dataframes=None):
                     )
 
                     if lti_is_dask or lti_is_spark:
-                        # if child_df.ww.name == "sessions":
-                        #     set_trace()
                         to_join = child_df[link_col]
                         if lti_is_dask:
                             to_join.index = child_df[child_df.ww.index]
@@ -1237,7 +1235,6 @@ def add_last_time_indexes(self, updated_dataframes=None):
                         lti_df = lti_df.groupby(lti_df[dataframe.ww.index]).agg("max")
 
                         lti_df = (
-                            # date
                             es_lti_dict[dataframe.ww.name]
                             .to_frame(name="last_time_old")
                             .join(lti_df)
@@ -1273,11 +1270,11 @@ def add_last_time_indexes(self, updated_dataframes=None):
                     else:
                         if lti_is_spark:
                             # TODO: Figure out a workaround for fillna and replace
-                            if lti_df["last_time_old"].dtype == "float64":
+                            if lti_df["last_time_old"].dtype != "datetime64[ns]":
                                 lti_df["last_time_old"] = ps.to_datetime(
                                     lti_df["last_time_old"],
                                 )
-                            if lti_df["last_time"].dtype == "float64":
+                            if lti_df["last_time"].dtype != "datetime64[ns]":
                                 lti_df["last_time"] = ps.to_datetime(
                                     lti_df["last_time"],
                                 )