Merge pull request #332 from astronomy-commons/update-margin-cache-info

Update margin cache catalog info parameters
astronomy-commons · Jun 6, 2024 · 3bcec06 · 3bcec06
2 parents 3b2c3ae + 87ecd85
commit 3bcec06
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+git+https://github.com/astronomy-commons/hipscat.git@main
diff --git a/src/hipscat_import/margin_cache/margin_cache_arguments.py b/src/hipscat_import/margin_cache/margin_cache_arguments.py
@@ -83,6 +83,9 @@ def to_catalog_info(self, total_rows) -> MarginCacheCatalogInfo:
             "catalog_name": self.output_artifact_name,
             "total_rows": total_rows,
             "catalog_type": "margin",
+            "epoch": self.catalog.catalog_info.epoch,
+            "ra_column": self.catalog.catalog_info.ra_column,
+            "dec_column": self.catalog.catalog_info.dec_column,
             "primary_catalog": self.input_catalog_path,
             "margin_threshold": self.margin_threshold,
         }

diff --git a/tests/hipscat_import/margin_cache/test_arguments_margin_cache.py b/tests/hipscat_import/margin_cache/test_arguments_margin_cache.py
@@ -125,6 +125,9 @@ def test_to_catalog_info(small_sky_source_catalog, tmp_path):
     catalog_info = args.to_catalog_info(total_rows=10)
     assert catalog_info.catalog_name == args.output_artifact_name
     assert catalog_info.total_rows == 10
+    assert catalog_info.epoch == "J2000"
+    assert catalog_info.ra_column == "source_ra"
+    assert catalog_info.dec_column == "source_dec"
 
 
 def test_provenance_info(small_sky_source_catalog, tmp_path):