Skip to content

added imports in layers file #79

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

Open
DeXtAr47-oss wants to merge 2 commits into
base: main
Choose a base branch
from
Open

added imports in layers file #79

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
771738e
added proper imports in layers and changed the code for merge_two_ar…
DeXtAr47-oss Nov 28, 2025
73b8a84
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 28, 2025
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
5 changes: 5 additions & 0 deletions metnet/data/merge_single_band.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ def merge_two_arrays(
sat_name_a: str = "gk2a",
sat_name_b: str = "goes_east",
sigma_deg: float = 20.0,
subpoints: Optional[Mapping[str, tuple]] = None,
) -> xr.DataArray:
"""
Merge two DataArrays on a common grid by aligning their coordinates.
Expand All @@ -99,6 +100,7 @@ def merge_two_arrays(
sat_name_a: satellite key for the first array (used by gaussian merge)
sat_name_b: satellite key for the second array (used by gaussian merge)
sigma_deg: Gaussian sigma in degrees (used by gaussian merge)
subpoints: optional dict mapping satellite names to (lat, lon) tuples for gaussian range

Returns:
A new DataArray containing merged values.
Expand All @@ -108,6 +110,9 @@ def merge_two_arrays(
if lat is None or lon is None:
raise ValueError("lat and lon grids are required for method='gaussian'.")

if subpoints is None:
subpoints = {sat_name_a: (0.0, 0.0), sat_name_b: (0.0, 0.0)}

merged = gaussian_merge_multi(
arrays=[satellite_one, satellite_two],
lat=lat,
Expand Down
13 changes: 13 additions & 0 deletions metnet/layers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,20 @@
"""Submodules for the layers."""

from .ConditionTime import ConditionTime
from .ConditionWithTimeMetNet2 import ConditionWithTimeMetNet2
from .ConvGRU import ConvGRU
from .ConvLSTM import ConvLSTM
from .CoordConv import CoordConv
from .DilatedCondConv import DilatedResidualConv
from .DownSampler import DownSampler
from .LeadTimeConditioner import LeadTimeConditioner
from .MaxViT import MetNetMaxVit
from .MBConv import MBConv
from .MultiheadSelfAttention2D import MultiheadSelfAttention2D
from .PartitionAttention import PartitionAttention
from .Preprocessor import MetNetPreprocessor
from .RelativePositionBias import RelativePositionBias
from .SqueezeExcitation import SqueezeExcite
from .StochasticDepth import StochasticDepth
from .TimeDistributed import TimeDistributed
from .utils import get_conv_layer