From 1df746bc869768271483c3145fe528845ed93cc5 Mon Sep 17 00:00:00 2001 From: Argenis Leon Date: Sat, 9 Nov 2019 17:14:03 -0600 Subject: [PATCH] Fix imports --- optimus/ml/models.py | 2 +- optimus/profiler/profiler.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/optimus/ml/models.py b/optimus/ml/models.py index dd901151..ac2bbf6c 100644 --- 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 index e67e8bcd..7ecbd634 100644 --- 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