Fix cluster plot title not being updated from config

VlachosGroup · Feb 15, 2022 · 24c2f1d · 24c2f1d
1 parent ba5f9e6
commit 24c2f1d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/AIMSim/tasks/cluster_data.py b/AIMSim/tasks/cluster_data.py
@@ -50,7 +50,6 @@ def _extract_configs(self):
             "embedding_plot_settings",
             {}))
 
-
         self.log_fpath = self.configs.get("log_file_path", None)
         if self.log_fpath is not None:
             log_dir = dirname(self.log_fpath)
@@ -141,7 +140,7 @@ def __call__(self, molecule_set):
             cluster_memberships=cluster_labels,
             xlabel=self.plot_settings["embedding_plot"]["xlabel"],
             ylabel=self.plot_settings["embedding_plot"]["ylabel"],
-            title=f"2-D projected space",
+            title=self.plot_settings["embedding_plot"]['plot_title'],
             hover_names=mol_names,
             cluster_colors=self.plot_settings["cluster_plot"]["cluster_colors"],
         )