Use display()to show the vis

VIDA-NYU · May 31, 2024 · 14853b5 · 14853b5
1 parent f75d411
commit 14853b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/bdikit/api.py b/bdikit/api.py
@@ -43,7 +43,7 @@ def load_dataset(self, dataset_path):
     def reduce_scope(self):
         self.scope_manager = ScopeReducingManager(self.dataset, self.global_table)
         self.reduced_scope = self.scope_manager.reduce()
-        return plot_reduce_scope(self.reduced_scope, self.dataset)
+        plot_reduce_scope(self.reduced_scope, self.dataset)
 
     def map_columns(self, algorithm='SimFloodAlgorithm'):
         self.column_manager = ColumnMappingManager(self.dataset, self.global_table, algorithm)

diff --git a/bdikit/visualization/mappings.py b/bdikit/visualization/mappings.py
@@ -7,13 +7,12 @@
 
 def plot_reduce_scope(reduced_scope, dataset):
     scope_explorer = ScopeReducerExplorer(dataset, reduced_scope)
-    return scope_explorer.explore()
+    display(scope_explorer.explore())
 
 def plot_column_mappings(column_mappings):
     column_mappings_df = pd.DataFrame(column_mappings.items(), columns=['Original Column', 'Target Column'])
     display(column_mappings_df)
 
-
 def plot_value_mappings(value_mappings, include_unmatches=True):
     sorted_results = sorted(value_mappings.items(), key=lambda x: x[1]['coverage'], reverse=True)