From 919b6781309dad532f3c202ffd85bb2913b84878 Mon Sep 17 00:00:00 2001 From: ArturoLlorente Date: Tue, 3 Oct 2023 10:06:18 +0200 Subject: [PATCH] changes werenot staged for commit --- src/graphnet/models/standard_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/graphnet/models/standard_model.py b/src/graphnet/models/standard_model.py index 467bfc462..25a6e5107 100644 --- 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]