From 7f99788f5be888627e53aff9ed2a732315e15c62 Mon Sep 17 00:00:00 2001 From: Simon Adamov Date: Sat, 25 May 2024 16:53:04 +0200 Subject: [PATCH] debug and linting --- neural_lam/utils.py | 3 +-- neural_lam/vis.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/neural_lam/utils.py b/neural_lam/utils.py index c6875aa3..59a718f4 100644 --- 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 index 2b6abf15..8c9ca77c 100644 --- 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