Added flake

Author: MatteoMazzonelli

.pre-commit-config.yaml

@@ -11,6 +11,11 @@ repos:
   hooks:
     - id: ruff-check
     - id: ruff-format
+-   repo: https://github.com/PyCQA/flake8
+    rev: 7.2.0
+    hooks:
+    -   id: flake8
+        args: ["--ignore=E501,W503,E203"]
 -   repo: https://github.com/asottile/add-trailing-comma
     rev: v3.2.0
     hooks:

gridFM/datasets/powergrid.py

@@ -89,7 +89,7 @@ def process(self):
         # Ensure node and edge data match
         assert (scenarios == edge_df["scenario"].unique()).all()
 
-        ## normalize node attributes
+        # normalize node attributes
         cols_to_normalize = ["Pd", "Qd", "Pg", "Qg", "Vm", "Va"]
         to_normalize = torch.tensor(
             node_df[cols_to_normalize].values,
@@ -100,7 +100,7 @@ def process(self):
             to_normalize,
         ).numpy()
 
-        ## normalize edge attributes
+        # normalize edge attributes
         cols_to_normalize = ["G", "B"]
         to_normalize = torch.tensor(
             edge_df[cols_to_normalize].values,
@@ -117,7 +117,7 @@ def process(self):
             to_normalize,
         ).numpy()
 
-        ## save stats
+        # save stats
         node_stats_path = osp.join(
             self.processed_dir,
             f"node_stats_{self.norm_method}.pt",
@@ -135,7 +135,7 @@ def process(self):
 
         data_list = []
         for scenario_idx in tqdm(scenarios):
-            ## NODE DATA
+            # NODE DATA
             node_data = node_groups.get_group(scenario_idx)
             x = torch.tensor(
                 node_data[
@@ -145,7 +145,7 @@ def process(self):
             )
             y = x[:, : self.mask_dim]
 
-            ## EDGE DATA
+            # EDGE DATA
             edge_data = edge_groups.get_group(scenario_idx)
             edge_attr = torch.tensor(edge_data[["G", "B"]].values, dtype=torch.float)
             edge_index = torch.tensor(