removing more asserts

sevmag · sevmag · commit b5ecc2b79cca · 2025-06-01T22:10:12.000+02:00
diff --git a/src/graphnet/models/data_representation/graphs/nodes/nodes.py b/src/graphnet/models/data_representation/graphs/nodes/nodes.py
@@ -645,9 +645,6 @@ def _construct_nodes(self, x: torch.Tensor) -> Data:
 
     def set_indeces(self, feature_names: List[str]) -> None:
         """Set the indices for the input features."""
-        assert set(feature_names) == set(
-            self._cluster_on + [self._charge_label, self._time_label]
-        ), f"Input feature names do not match: {feature_names}"
         self._cluster_idx = [
             feature_names.index(column) for column in self._cluster_on
         ]

Original file line number	Diff line number	Diff line change
`@@ -645,9 +645,6 @@ def _construct_nodes(self, x: torch.Tensor) -> Data:`
`645`	`645`
`646`	`646`	`def set_indeces(self, feature_names: List[str]) -> None:`
`647`	`647`	`"""Set the indices for the input features."""`
`648`		`- assert set(feature_names) == set(`
`649`		`- self._cluster_on + [self._charge_label, self._time_label]`
`650`		`- ), f"Input feature names do not match: {feature_names}"`
`651`	`648`	`self._cluster_idx = [`
`652`	`649`	`feature_names.index(column) for column in self._cluster_on`
`653`	`650`	`]`