syntax

splicemachine · Jan 12, 2021 · f0b54fe · f0b54fe
1 parent 3d5e099
commit f0b54fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/splicemachine/features/feature_store.py b/splicemachine/features/feature_store.py
@@ -333,7 +333,7 @@ def get_training_set(self, features: Union[List[Feature], List[str]], current_va
         if current_values_only:
             ts.start_time = ts.end_time
 
-        if hasattr(self, 'mlflow_ctx' and not return_sql):
+        if hasattr(self, 'mlflow_ctx') and not return_sql:
             self.mlflow_ctx._active_training_set: TrainingSet = ts
             ts._register_metadata(self.mlflow_ctx)
         return sql if return_sql else self.splice_ctx.df(sql)