DRY code:

logicalclocks · Oct 29, 2020 · 555ada7 · 555ada7
1 parent 3fea11d
commit 555ada7
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/python/hsfs/training_dataset.py b/python/hsfs/training_dataset.py
@@ -151,16 +151,14 @@ def save(
         if isinstance(features, query.Query):
             feature_dataframe = features.read("offline")
             self._querydto = features
-            self._features = engine.get_instance().parse_schema_training_dataset(
-                feature_dataframe
-            )
         else:
             feature_dataframe = engine.get_instance().convert_to_default_dataframe(
                 features
             )
-            self._features = engine.get_instance().parse_schema_training_dataset(
-                feature_dataframe
-            )
+
+        self._features = engine.get_instance().parse_schema_training_dataset(
+            feature_dataframe
+        )
 
         self._set_label_features()