attempt to modify

mpl_data_plotter.py

@@ -14,17 +14,17 @@ class MatplotlibDataPlotter:
 
         self.num_domains_in_region_df = num_domains_in_region_df
 
-        self.single_domains_fig = plt.figure(figsize=(5, 10))
-        self.pair_domains_fig = plt.figure(figsize=(5, 10))
+        #self.single_domains_fig = plt.figure(figsize=(5, 10))
+        #self.pair_domains_fig = plt.figure(figsize=(5, 10))
 
     def plot_single_domains(self, num_domains, split_name):
-        
+        return plt.gcf()
         # selected_region_ids = self.num_domains_in_region_df.loc[
         #     self.num_domains_in_region_df.num_domains >= num_domains, 
         #     'cds_region_id'].values
         # single_df_subset = self.single_df.loc[self.single_df.cds_region_id.isin(selected_region_ids)]
         
-        return self.single_domains_fig
+        
         # split_name = 'stratified'
         column_name = f'cosine_similarity_{split_name}'
         # single_df_subset = single_df.loc[single_df.dom_location_len >= num_domains]
@@ -64,11 +64,11 @@ class MatplotlibDataPlotter:
         return fig # plt.gcf()
 
     def plot_pair_domains(self, num_domains, split_name):
+        return plt.gcf()
         # selected_region_ids = self.num_domains_in_region_df.loc[
         #     self.num_domains_in_region_df.num_domains >= num_domains, 
         #     'cds_region_id'].values
         # pair_df_subset = self.pair_df.loc[self.pair_df.cds_region_id.isin(selected_region_ids)]
-        return self.pair_domains_fig
         # split_name = 'stratified'
         column_name = f'cosine_similarity_{split_name}'
         # pair_df_subset = pair_df.loc[pair_df.dom_location_len >= num_domains]