Fix tests

eapframework · Dec 6, 2019 · acd8d5f · acd8d5f
1 parent 820640a
commit acd8d5f
Show file tree

Hide file tree

Showing 2 changed files with 363 additions and 23 deletions.
diff --git a/optimus/dataframe/rows.py b/optimus/dataframe/rows.py
@@ -6,7 +6,7 @@
 from pyspark.sql import functions as F
 
 # Helpers
-import optimus.optimus as op
+from optimus import Optimus as op
 from infer import is_list_of_str_or_int, is_list_of_dataframes, is_list_of_tuples, is_dataframe, \
     PYSPARK_NUMERIC_TYPES
 from optimus.audf import filter_row_by_data_type as fbdt