diff --git a/AIMSim/tasks/cluster_data.py b/AIMSim/tasks/cluster_data.py index 2937c573..2a583bb2 100644 --- 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"], )