Fix imports

eapframework · Nov 9, 2019 · 1df746b · 1df746b
1 parent 3d7e984
commit 1df746b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/optimus/ml/models.py b/optimus/ml/models.py
@@ -1,6 +1,6 @@
 from pyspark.ml import feature, classification
 from pyspark.ml.classification import RandomForestClassifier, DecisionTreeClassifier, GBTClassifier
-from pyspark.sql.functions import F
+from pyspark.sql import functions as F
 from pysparkling import *
 from pysparkling.ml import H2OAutoML, H2ODeepLearning, H2OXGBoost, H2OGBM
 

diff --git a/optimus/profiler/profiler.py b/optimus/profiler/profiler.py
@@ -13,15 +13,14 @@
 from optimus.helpers.check import is_column_a, is_dict, is_list_of_str
 from optimus.helpers.columns import parse_columns
 from optimus.helpers.columns_expression import zeros_agg, count_na_agg, hist_agg, percentile_agg, count_uniques_agg
-from optimus.helpers.constants import RELATIVE_ERROR, Actions
+from optimus.helpers.constants import RELATIVE_ERROR, Actions, PYSPARK_NUMERIC_TYPES
 from optimus.helpers.decorators import time_it
 from optimus.helpers.functions import absolute_path
 from optimus.helpers.json import json_converter
 from optimus.helpers.logger import logger
 from optimus.helpers.output import print_html
 from optimus.helpers.raiseit import RaiseIt
-from optimus.profiler.functions import fill_missing_col_types, \
-    write_json, write_html, PYSPARK_NUMERIC_TYPES
+from optimus.profiler.functions import fill_missing_col_types, write_json, write_html
 from optimus.profiler.templates.html import FOOTER, HEADER
 
 MAX_BUCKETS = 33