Merge pull request #29 from microsoft/wesselb/remove-redundant-lines

Remove redundant line in `aurora.py`
microsoft · Sep 10, 2024 · 8c2f0c2 · 8c2f0c2
2 parents ffeadc4 + 9b449c9
commit 8c2f0c2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py
@@ -198,7 +198,6 @@ def forward(self, batch: Batch) -> Batch:
         )
 
         # Remove batch and history dimension from static variables.
-        B, T = next(iter(batch.surf_vars.values()))[0]
         pred = dataclasses.replace(
             pred,
             static_vars={k: v[0, 0] for k, v in batch.static_vars.items()},