Revert "[HOPSWORKS-3201][append] set training dataset start end time (#…

…679)" This reverts commit 8d37269.
logicalclocks · Jun 16, 2022 · df4f81d · df4f81d
1 parent 6ccde4f
commit df4f81d
Showing 1 changed file with 4 additions and 20 deletions.
diff --git a/python/hsfs/core/feature_view_engine.py b/python/hsfs/core/feature_view_engine.py
@@ -227,34 +227,18 @@ def _set_event_time(self, feature_view_obj, training_dataset_obj):
                         split.name == TrainingDatasetSplit.TRAIN
                         and not split.start_time
                     ):
-                        df = (
-                            feature_view_obj.query._left_feature_group.select_all().read()
-                            if df is None
-                            else df
-                        )
+                        df = feature_view_obj.query.read() if df is None else df
                         split.start_time = self._get_start_time(df, event_time)
                     if split.name == TrainingDatasetSplit.TEST and not split.end_time:
-                        df = (
-                            feature_view_obj.query._left_feature_group.select_all().read()
-                            if df is None
-                            else df
-                        )
+                        df = feature_view_obj.query.read() if df is None else df
                         split.end_time = self._get_end_time(df, event_time)
             else:
                 if not training_dataset_obj.event_start_time:
-                    df = (
-                        feature_view_obj.query._left_feature_group.select_all().read()
-                        if df is None
-                        else df
-                    )
+                    df = feature_view_obj.query.read() if df is None else df
                     training_dataset_obj.event_start_time = self._get_start_time(
                         df, event_time
                     )
-                    df = (
-                        feature_view_obj.query._left_feature_group.select_all().read()
-                        if df is None
-                        else df
-                    )
+                    df = feature_view_obj.query.read() if df is None else df
                     training_dataset_obj.event_end_time = self._get_end_time(
                         df, event_time
                     )