From e76bdb25352413ea70968a702edd30fc897a6d64 Mon Sep 17 00:00:00 2001
From: Altay Sansal <13684161+tasansal@users.noreply.github.com>
Date: Mon, 20 Oct 2025 09:58:09 -0500
Subject: [PATCH] remove "gun" logical coordinate from seismic 2d prestack shot
 template and tests

---
 .../templates/seismic_2d_prestack_shot.py       |  7 -------
 .../templates/test_seismic_2d_prestack_shot.py  | 17 ++++-------------
 2 files changed, 4 insertions(+), 20 deletions(-)

diff --git a/src/mdio/builder/templates/seismic_2d_prestack_shot.py b/src/mdio/builder/templates/seismic_2d_prestack_shot.py
index 62ea293e..6463ac77 100644
--- a/src/mdio/builder/templates/seismic_2d_prestack_shot.py
+++ b/src/mdio/builder/templates/seismic_2d_prestack_shot.py
@@ -17,7 +17,6 @@ def __init__(self, data_domain: SeismicDataDomain):
 
         self._dim_names = ("shot_point", "channel", self._data_domain)
         self._physical_coord_names = ("source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y")
-        self._logical_coord_names = ("gun",)
         self._var_chunk_shape = (16, 32, 2048)
 
     @property
@@ -39,12 +38,6 @@ def _add_coordinates(self) -> None:
 
         # Add non-dimension coordinates
         compressor = compressors.Blosc(cname=compressors.BloscCname.zstd)
-        self._builder.add_coordinate(
-            "gun",
-            dimensions=("shot_point",),
-            data_type=ScalarType.UINT8,
-            compressor=compressor,
-        )
         self._builder.add_coordinate(
             "source_coord_x",
             dimensions=("shot_point",),
diff --git a/tests/unit/v1/templates/test_seismic_2d_prestack_shot.py b/tests/unit/v1/templates/test_seismic_2d_prestack_shot.py
index bd494959..1ec90fe6 100644
--- a/tests/unit/v1/templates/test_seismic_2d_prestack_shot.py
+++ b/tests/unit/v1/templates/test_seismic_2d_prestack_shot.py
@@ -23,14 +23,14 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
     """Validate the coordinate, headers, trace_mask variables in the dataset."""
     # Verify variables
     # 4 dim coords + 5 non-dim coords + 1 data + 1 trace mask + 1 headers = 12 variables
-    assert len(dataset.variables) == 11
+    assert len(dataset.variables) == 10
 
     # Verify trace headers
     validate_variable(
         dataset,
         name="headers",
         dims=[("shot_point", 256), ("channel", 24)],
-        coords=["gun", "source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
+        coords=["source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
         dtype=headers,
     )
 
@@ -38,7 +38,7 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
         dataset,
         name="trace_mask",
         dims=[("shot_point", 256), ("channel", 24)],
-        coords=["gun", "source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
+        coords=["source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
         dtype=ScalarType.BOOL,
     )
 
@@ -69,14 +69,6 @@ def _validate_coordinates_headers_trace_mask(dataset: Dataset, headers: Structur
     assert domain.metadata.units_v1 == UNITS_SECOND
 
     # Verify non-dimension coordinate variables
-    validate_variable(
-        dataset,
-        name="gun",
-        dims=[("shot_point", 256)],
-        coords=["gun"],
-        dtype=ScalarType.UINT8,
-    )
-
     source_coord_x = validate_variable(
         dataset,
         name="source_coord_x",
@@ -125,7 +117,6 @@ def test_configuration(self) -> None:
         assert t._data_domain == "time"
         assert t._dim_names == ("shot_point", "channel", "time")
         assert t._physical_coord_names == ("source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y")
-        assert t._logical_coord_names == ("gun",)
         assert t.full_chunk_shape == (16, 32, 2048)
 
         # Variables instantiated when build_dataset() is called
@@ -159,7 +150,7 @@ def test_build_dataset(self, structured_headers: StructuredType) -> None:
             dataset,
             name="amplitude",
             dims=[("shot_point", 256), ("channel", 24), ("time", 2048)],
-            coords=["gun", "source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
+            coords=["source_coord_x", "source_coord_y", "group_coord_x", "group_coord_y"],
             dtype=ScalarType.FLOAT32,
         )
         assert isinstance(seismic.compressor, Blosc)