Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 12, 2024
1 parent dad5d11 commit 65cc6fd
Show file tree
Hide file tree
Showing 12 changed files with 12 additions and 0 deletions.
1 change: 1 addition & 0 deletions metnet/layers/ConditionWithTimeMetNet2.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Condition with time how MetNet-22 does it, with FiLM layers"""

import einops
import torch
from torch import nn as nn
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/ConvLSTM.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Originally adapted from https://github.com/aserdega/convlstmgru, MIT License Andriy Serdega"""

from typing import Any, List, Optional

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/DilatedCondConv.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Dilated Time Conditioned Residual Convolution Block for MetNet-2"""

import torch
import torch.nn as nn
import torch.nn.functional as F
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/MBConv.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
MBConv Implementation
"""

from typing import Type

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/MaxViT.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Implementation of MaxViT module
"""

from dataclasses import dataclass
from typing import List, Tuple, Type, Union

Expand Down
1 change: 1 addition & 0 deletions metnet/layers/MultiheadSelfAttention2D.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
MultiHeaded 2D Self Attention Implementation
"""

from typing import Type

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/PartitionAttention.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Implementation of Partition (Grid and Block) Attention
"""

from typing import Tuple, Type

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/RelativePositionBias.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Implementation of Relative Position Bias
"""

from typing import Tuple

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/SqueezeExcitation.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Squeeze-and-Excitation net implementation
"""

from typing import Type

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/layers/StochasticDepth.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Implementation of Stochastic Depth
"""

import torch
from torch import nn

Expand Down
1 change: 1 addition & 0 deletions metnet/models/metnet2.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""MetNet-2 model for weather forecasting"""

from typing import List

import torch
Expand Down
1 change: 1 addition & 0 deletions metnet/utils.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Utilities"""

import numpy as np


Expand Down

0 comments on commit 65cc6fd

Please sign in to comment.