Fix display function for analyzer

hi-primus · Oct 7, 2017 · 76c07da · 76c07da
1 parent 224303d
commit 76c07da
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/optimus/df_analyzer.py b/optimus/df_analyzer.py
@@ -817,16 +817,15 @@ def write_json(cls, json_cols, path_to_json_file):
             # outfile.write(str(json_cols).replace("'", "\""))
             outfile.write(json_cols)
 
-    @classmethod
-    def display_optimus(cls, df):
+    def display_optimus(self):
         """
         Lets you visualize your Spark object in different ways: table, charts, maps, etc.
         :param df: spark dataframe to be analyzed
         :return: Amazing visualizations for your spark dataframes.
         """
         # Import pixiedust-optimus
         from pixiedust_optimus.display import display
-        display(df)
+        display(self._df)
 
     def get_frequency(self, columns, sort_by_count=True):
         """