Improves performance of QL by changing how the QL extracts node data …

…when using MultiIndex (#143)
LLNL · Aug 26, 2024 · 4375612 · 4375612
1 parent 30f0758
commit 4375612
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hatchet/query/engine.py b/hatchet/query/engine.py
@@ -75,7 +75,7 @@ def _cache_node(self, node, query, dframe):
             _, filter_func = node_query
             row = None
             if isinstance(dframe.index, pd.MultiIndex):
-                row = pd.concat([dframe.loc[node]], keys=[node], names=["node"])
+                row = dframe.xs(node, level="node", drop_level=False)
             else:
                 row = dframe.loc[node]
             if filter_func(row):