removing unnecessary functionality

Author: sevmag

src/graphnet/models/data_representation/images/image_representation.py

@@ -77,29 +77,6 @@ def shape(self) -> List[List[int]]:
         """Channel-spatial layout per image tensor (see ``GridDefinition``)."""
         return self._grid_definition.shape
 
-    def single_image_spatial_shape(self) -> Tuple[int, int, int]:
-        """Return spatial size as ``(height, width, depth)`` for one 3D image.
-
-        Raises:
-            ValueError: If ``shape`` does not describe exactly one four-axis
-                layout (channels plus three spatial axes).
-        """
-        layouts = self.shape
-        if len(layouts) != 1:
-            raise ValueError(
-                "Expected a single-image data representation (one shape "
-                f"entry), got {len(layouts)}. For multi-image inputs, build "
-                "the backbone explicitly for each tensor."
-            )
-        layout = layouts[0]
-        if len(layout) != 4:
-            raise ValueError(
-                "Expected each image layout as "
-                "[num_channels, height, width, depth]; "
-                f"got {layout!r}."
-            )
-        return (layout[1], layout[2], layout[3])
-
     def _set_output_feature_names(
         self, input_feature_names: List[str]
     ) -> List[str]:

src/graphnet/models/data_representation/images/mappings/grid_definition.py

@@ -100,11 +100,6 @@ def __init__(
     ) -> None:
         """Construct `IC86GridDefinition`.
 
-        The mapping from (string, dom_number) to a position in the
-        resulting images is generated programmatically at instantiation
-        time from the IceCube86 detector geometry, so no auxiliary file
-        is required.
-
         Args:
             detector: ``IceCube86`` instance (grid is fixed to that geometry).
             dtype: data type used for node features. e.g. ´torch.float´
@@ -169,6 +164,8 @@ def _set_indices(
     ) -> None:
         """Set the indices for the features."""
         self._cnn_features_idx = []
+        self._dom_number_idx = None
+        self._string_idx = None
         for feature in feature_names:
             if feature == dom_number_label:
                 self._dom_number_idx = feature_names.index(feature)
@@ -177,6 +174,15 @@ def _set_indices(
             else:
                 self._cnn_features_idx.append(feature_names.index(feature))
 
+        if self._dom_number_idx is None:
+            raise ValueError(
+                f"DOM number label not found in feature names: {dom_number_label}"
+            )
+        if self._string_idx is None:
+            raise ValueError(
+                f"String label not found in feature names: {string_label}"
+            )
+
     def forward(self, data: Data, data_feature_names: List[str]) -> Data:
         """Scatter pixel rows into IceCube-86 image tensor(s)."""
         # Initialize output arrays