Skip to content

Bump torch-harmonics to 0.8.0 and update constraint on imageio #49

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Nov 4, 2025
Merged

Bump torch-harmonics to 0.8.0 and update constraint on imageio #49

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions fme/ace/aggregator/inference/video.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -509,9 +509,13 @@ def _make_video(
video_data = np.minimum(video_data, 255)
video_data = np.maximum(video_data, 0)
video_data[np.isnan(video_data)] = 0
# Convert from single-channel grayscale to three-channel RGB to work
# around imageio error
video_data = video_data.repeat(3, axis=1)
caption += f"; vmin={data_min:.4g}, vmax={data_max:.4g}"
return wandb.Video(
np.flip(video_data, axis=-2),
caption=caption,
fps=4,
format="gif",
)
8 changes: 3 additions & 5 deletions fme/sht_fix.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#

import numpy as np
import torch
import torch.nn as nn
import torch.fft
Expand Down Expand Up @@ -100,15 +99,14 @@ def __init__(self, nlat, nlon, lmax=None, mmax=None, grid="lobatto", norm="ortho
raise(ValueError("Unknown quadrature mode"))

# apply cosine transform and flip them
tq = np.flip(np.arccos(cost))
tq = torch.flip(torch.arccos(cost), dims=(0,))

# determine the dimensions
self.mmax = mmax or self.nlon // 2 + 1

# combine quadrature weights with the legendre weights
weights = torch.from_numpy(w)
pct = torch.as_tensor(_precompute_legpoly(self.mmax, self.lmax, tq, norm=self.norm, csphase=self.csphase))
weights = torch.einsum('mlk,k->mlk', pct, weights)
weights = torch.einsum('mlk,k->mlk', pct, w)

# remember quadrature weights
self.weights = weights.float().to(get_device())
Expand Down Expand Up @@ -181,7 +179,7 @@ def __init__(self, nlat, nlon, lmax=None, mmax=None, grid="lobatto", norm="ortho
raise(ValueError("Unknown quadrature mode"))

# apply cosine transform and flip them
t = np.flip(np.arccos(cost))
t = torch.flip(torch.arccos(cost), dims=(0,))

# determine the dimensions
self.mmax = mmax or self.nlon // 2 + 1
Expand Down
4 changes: 2 additions & 2 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ dacite
gcsfs
h5netcdf
h5py
imageio<=2.27.0
imageio>=2.28.1
matplotlib
moviepy
netcdf4
Expand All @@ -14,7 +14,7 @@ plotly
s3fs
tensorly
tensorly-torch
torch-harmonics==0.7.4 # pinned since we use private API
torch-harmonics==0.8.0 # pinned since we use private API
torch>=2
wandb[media]>=0.19.0
xarray
Expand Down