?

Author: ManuelLecha

modules/transforms/feature_liftings/feature_liftings.py

@@ -28,9 +28,7 @@ def lift_features(
         -------
         torch_geometric.data.Data | dict
             The lifted data."""
-        keys = sorted(
-            [key.split("_")[1] for key in data if "incidence" in key]
-        )  # : SIM118
+        keys = sorted([key.split("_")[1] for key in data.keys() if "incidence" in key])  # noqa : SIM118
         for elem in keys:
             if f"x_{elem}" not in data:
                 idx_to_project = 0 if elem == "hyperedges" else int(elem) - 1
@@ -55,4 +53,4 @@ def forward(
         torch_geometric.data.Data | dict
             The lifted data.
         """
-        return self.lift_features(data)
+        return self.lift_features(data)

test/transforms/feature_liftings/test_projection_sum.py

@@ -87,4 +87,4 @@ def test_lift_features(self):
 
         assert (
             expected_x_hyperedges == lifted_data.x_hyperedges
-        ).all(), "Something is wrong with the lifted features x_hyperedges."
+        ).all(), "Something is wrong with the lifted features x_hyperedges."