Better parameter filtering

xpublish-community · Nov 21, 2024 · 3a625ea · 3a625ea
1 parent b8970f7
commit 3a625ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/test_cf_router.py b/tests/test_cf_router.py
@@ -108,6 +108,10 @@ def test_cf_metadata_query(cf_client):
     ], "Spatial bbox is incorrect"
     assert data["extent"]["spatial"]["crs"] == "EPSG:4326", "Spatial CRS is incorrect"
 
+    assert "air" in data["parameter_names"], "Air parameter should be present"
+    assert "lat" not in data["parameter_names"], "lat should not be present"
+    assert "lon" not in data["parameter_names"], "lon should not be present"
+
 
 def test_cf_metadata_query_temp_smoke_test(cf_client):
     response = cf_client.get("/datasets/temp/edr/")

diff --git a/xpublish_edr/plugin.py b/xpublish_edr/plugin.py
@@ -177,7 +177,7 @@ def get_collection_metadata(dataset: xr.Dataset = Depends(deps.dataset)):
 
             parameters = {
                 k: variable_description(v)
-                for k, v in dataset.variables.items()
+                for k, v in projected_ds.variables.items()
                 if "axis" not in v.attrs
             }