From 555ada789c1e242e0d2f0ec1558776bea29f38c4 Mon Sep 17 00:00:00 2001 From: moritzmeister Date: Thu, 29 Oct 2020 14:17:54 +0100 Subject: [PATCH] DRY code: --- python/hsfs/training_dataset.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/python/hsfs/training_dataset.py b/python/hsfs/training_dataset.py index 8f0422d216..9621a197b7 100644 --- 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()