changes werenot staged for commit

graphnet-team · Oct 3, 2023 · 919b678 · 919b678
1 parent 4f04ffc
commit 919b678
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/graphnet/models/standard_model.py b/src/graphnet/models/standard_model.py
@@ -94,7 +94,9 @@ def configure_optimizers(self) -> Dict[str, Any]:
             )
         return config
 
-    def forward(self, data: Union[Data, List[Data]]) -> List[Union[Tensor, Data]]:
+    def forward(
+        self, data: Union[Data, List[Data]]
+    ) -> List[Union[Tensor, Data]]:
         """Forward pass, chaining model components."""
         if isinstance(data, Data):
             data = [data]