Fix formatting inconsistencies

Author: FilippoOlivo

pina/condition/condition_interface.py

@@ -12,4 +12,4 @@ def problem(self):
 
     @problem.setter
     def problem(self, value):
-        self._problem = value
+        self._problem = value

pina/data/dataset.py

@@ -195,7 +195,7 @@ def _getitem_dummy(self, idx):
 
     def _getitem_int(self, idx):
         return {
-            k: v['graph'][idx % len(v["graph"])]
+            k: v["graph"][idx % len(v["graph"])]
             for k, v in self.conditions_dict.items()
         }
 
@@ -234,7 +234,6 @@ def create_graph_batch(self, data):
         """
         print(data)
         return self._create_graph_batch_from_list(data)
-    
 
     @staticmethod
     def _extract_output(func):

pina/solver/supervised.py

@@ -37,7 +37,10 @@ class SupervisedSolver(SingleSolverInterface):
     multiple (discretised) input functions.
     """
 
-    accepted_conditions_types = (InputOutputPointsCondition, GraphInputOutputCondition)
+    accepted_conditions_types = (
+        InputOutputPointsCondition,
+        GraphInputOutputCondition,
+    )
 
     def __init__(
         self,

tests/test_graph.py

@@ -5,7 +5,7 @@
 from torch_geometric.data import Data
 
 
-def build_edge_attr(graph ):
+def build_edge_attr(graph):
     pos = graph.pos
     edge_index = graph.edge_index
     return torch.cat([pos[edge_index[0]], pos[edge_index[1]]], dim=-1)

tests/test_problem_zoo/test_supervised_problem.py

@@ -24,7 +24,7 @@ def test_constructor_graph():
         RadiusGraph(x=x_, pos=pos_, radius=0.2, edge_attr=True, y=y_)
         for x_, pos_, y_ in zip(x, pos, output_)
     ]
-    
+
     problem = SupervisedProblem(graph_=input_)
     assert isinstance(problem, AbstractProblem)
     assert hasattr(problem, "conditions")