Ensure mrid is index in switch dataframe

statnett · Apr 12, 2024 · 9504aa4 · 9504aa4
1 parent fdc694d
commit 9504aa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cimsparql/model.py b/cimsparql/model.py
@@ -663,7 +663,7 @@ def switches_query(self) -> str:
 
     @time_it
     def switches(self) -> SwitchesDataFrame:
-        df = self.get_table_and_convert(self.switches_query())
+        df = self.get_table_and_convert(self.switches_query(), index="mrid")
         return SwitchesDataFrame(df)
 
     def connectivity_nodes_query(self, region: str | None = None) -> str: