fix idx error

graphnet-team · Oct 27, 2023 · 035b81b · 035b81b
1 parent 6d46fc5
commit 035b81b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/graphnet/models/graphs/graph_definition.py b/src/graphnet/models/graphs/graph_definition.py
@@ -274,9 +274,7 @@ def _attach_inactive_sensors(
         unique_sensors = geometry_table.reset_index(drop=True)
 
         # multiple lines to avoid long line:
-        inactive_idx = ~geometry_table.isin(lookup).reset_index(drop=True)
-        inactive_idx = inactive_idx[geometry_table.columns[0]]
-        #
+        inactive_idx = ~geometry_table.index.isin(lookup)
         inactive_sensors = unique_sensors.loc[
             inactive_idx, input_feature_names
         ]