more files

Signed-off-by: Thijs Baaijen <[email protected]>

Author: Thijss

src/power_grid_model_ds/_core/data_source/generator/arrays/node.py

@@ -16,22 +16,22 @@ class NodeGenerator(BaseGenerator):
     def run(self, amount: int, voltage_level: int = 10_500):
         """Generate nodes in a grid with two possible load scenarios"""
         node_array = self.grid.node.__class__.zeros(amount)
-        node_array.id = 1 + self.grid.max_id + np.arange(amount)
-        node_array.u_rated = voltage_level
+        node_array["id"] = 1 + self.grid.max_id + np.arange(amount)
+        node_array["u_rated"] = voltage_level
 
         load_low_array = self.grid.sym_load.__class__.zeros(amount)
-        load_low_array.id = 1 + node_array.id.max() + np.arange(amount)
-        load_low_array.node = node_array.id
-        load_low_array.status = 1
+        load_low_array["id"] = 1 + node_array["id"].max() + np.arange(amount)
+        load_low_array["node"] = node_array["id"]
+        load_low_array["status"] = 1
         load_high_array = self.grid.sym_load.__class__.zeros(amount)
-        load_high_array.id = 1 + load_low_array.id.max() + np.arange(amount)
-        load_high_array.node = node_array.id
-        load_high_array.status = 1
+        load_high_array["id"] = 1 + load_low_array["id"].max() + np.arange(amount)
+        load_high_array["node"] = node_array["id"]
+        load_high_array["status"] = 1
 
         # power consumption in Watt
-        load_low_array.p_specified = np.round(self.rng.normal(200_000, 150_000, amount))
-        load_low_array.q_specified = np.round(self.rng.normal(20_000, 15_000, amount))
-        load_high_array.p_specified = np.round(self.rng.normal(-100_000, 350_000, amount))
-        load_high_array.q_specified = np.round(self.rng.normal(-5_000, 35_000, amount))
+        load_low_array["p_specified"] = np.round(self.rng.normal(200_000, 150_000, amount))
+        load_low_array["q_specified"] = np.round(self.rng.normal(20_000, 15_000, amount))
+        load_high_array["p_specified"] = np.round(self.rng.normal(-100_000, 350_000, amount))
+        load_high_array["q_specified"] = np.round(self.rng.normal(-5_000, 35_000, amount))
 
         return node_array, load_low_array, load_high_array

src/power_grid_model_ds/_core/visualizer/parsers.py

@@ -18,10 +18,11 @@ def parse_node_array(nodes: NodeArray) -> list[dict[str, Any]]:
     columns = nodes.columns
     for node in nodes:
         cyto_elements = {"data": _array_to_dict(node, columns)}
-        cyto_elements["data"]["id"] = str(node.id.item())
+        cyto_elements["data"]["id"] = str(node["id"].item())
         cyto_elements["data"]["group"] = "node"
         if with_coords:
-            cyto_elements["position"] = {"x": node.x.item(), "y": -node.y.item()}  # invert y-axis for visualization
+            y_value = -node["y"].item()  # invert y-axis for visualization
+            cyto_elements["position"] = {"x": node["x"].item(), "y": y_value}
         parsed_nodes.append(cyto_elements)
     return parsed_nodes
 
@@ -46,9 +47,9 @@ def parse_branch3_array(branches: Branch3Array, group: Literal["transformer"]) -
             cyto_elements["data"].update(
                 {
                     # IDs need to be unique, so we combine the branch ID with the from and to nodes
-                    "id": str(branch3.id.item()) + f"_{branch1.from_node.item()}_{branch1.to_node.item()}",
-                    "source": str(branch1.from_node.item()),
-                    "target": str(branch1.to_node.item()),
+                    "id": str(branch3["id"].item()) + f"_{branch1['from_node'].item()}_{branch1['to_node'].item()}",
+                    "source": str(branch1["from_node"].item()),
+                    "target": str(branch1["to_node"].item()),
                     "group": group,
                 }
             )
@@ -64,9 +65,9 @@ def parse_branch_array(branches: BranchArray, group: Literal["line", "link", "tr
         cyto_elements = {"data": _array_to_dict(branch, columns)}
         cyto_elements["data"].update(
             {
-                "id": str(branch.id.item()),
-                "source": str(branch.from_node.item()),
-                "target": str(branch.to_node.item()),
+                "id": str(branch["id"].item()),
+                "source": str(branch["from_node"].item()),
+                "target": str(branch["to_node"].item()),
                 "group": group,
             }
         )