diff --git a/nuztf/base_scanner.py b/nuztf/base_scanner.py index 2c41f62..e320ca0 100644 --- a/nuztf/base_scanner.py +++ b/nuztf/base_scanner.py @@ -209,9 +209,9 @@ def add_res_to_cache(self, res): ): self.cache[res_alert["objectId"]] = res_alert - def add_to_cache_by_names(self, *args): - for ztf_name in args: - query_res = ampel_api_name(ztf_name, logger=self.logger) + def add_to_cache_by_names(self, ztf_ids: list): + for ztf_id in ztf_ids: + query_res = ampel_api_name(ztf_id, logger=self.logger) self.add_res_to_cache(query_res) def check_ampel_filter(self, ztf_name): @@ -551,7 +551,7 @@ def create_candidate_summary(self): pdf_path = self.get_output_dir() / "candidates.pdf" self.logger.info( - f"Creating overview pdf\n(this might take a moment as it involves catalog matching)" + f"Creating overviewpdf\n(this might take a moment as it involves catalog matching)" ) self.logger.debug(f"Overview pdf path: {pdf_path}") diff --git a/pyproject.toml b/pyproject.toml index 5e81169..afc4959 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "nuztf" -version = "2.7.1" +version = "2.7.2" description = "Package for multi-messenger correlation searches with ZTF" authors = ["Robert Stein ", "Simeon Reusch ", "Jannis Necker "] repository = "https://github.com/desy-multimessenger/nuztf" diff --git a/tests/test_neutrino_scanner.py b/tests/test_neutrino_scanner.py index 385daf1..a0168c1 100644 --- a/tests/test_neutrino_scanner.py +++ b/tests/test_neutrino_scanner.py @@ -118,7 +118,7 @@ def test_scan(self): # Test manually adding candidates nu.add_to_cache_by_names( - "ZTF18abteipt", + ztf_ids=["ZTF18abteipt"], ) # Check