debug and linting

MeteoSwiss · May 25, 2024 · 7f99788 · 7f99788
1 parent 0ed609a
commit 7f99788
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/neural_lam/utils.py b/neural_lam/utils.py
@@ -7,12 +7,11 @@
 import numpy as np
 import pytorch_lightning as pl
 import torch
+import wandb  # pylint: disable=wrong-import-order
 from pytorch_lightning.utilities import rank_zero_only
 from torch import nn
 from tueplots import bundles, figsizes
 
-import wandb  # pylint: disable=wrong-import-order
-
 
 def load_dataset_stats(dataset_name, device="cpu"):
     """

diff --git a/neural_lam/vis.py b/neural_lam/vis.py
@@ -87,7 +87,7 @@ def plot_prediction(
         1,
         2,
         figsize=(13, 7),
-        subplot_kw={"projection": data_config.coords_projection()},
+        subplot_kw={"projection": data_config.coords_projection},
     )
 
     # Plot pred and target
@@ -136,7 +136,7 @@ def plot_spatial_error(error, obs_mask, data_config, title=None, vrange=None):
 
     fig, ax = plt.subplots(
         figsize=(5, 4.8),
-        subplot_kw={"projection": data_config.coords_projection()},
+        subplot_kw={"projection": data_config.coords_projection},
     )
 
     ax.coastlines()  # Add coastline outlines