Rename files

microsoft · Jul 30, 2024 · 5d8900e · 5d8900e
1 parent 0d1b855
commit 5d8900e
Show file tree

Hide file tree

Showing 8 changed files with 6 additions and 171 deletions.
diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py
@@ -6,9 +6,9 @@
 from torch import nn
 
 from aurora.batch import Batch
-from aurora.model.perceiver_decoder import Perceiver3DDecoder
-from aurora.model.perceiver_encoder import Perceiver3DEncoder
-from aurora.model.swin_3d_block import Int3Tuple, Swin3DTransformerBackbone
+from aurora.model.decoder import Perceiver3DDecoder
+from aurora.model.encoder import Perceiver3DEncoder
+from aurora.model.swin3d import Int3Tuple, Swin3DTransformerBackbone
 
 __all__ = ["Aurora"]
 

diff --git a/aurora/model/perceiver_decoder.py → aurora/model/decoder.py b/aurora/model/perceiver_decoder.py → aurora/model/decoder.py
diff --git a/aurora/model/perceiver_encoder.py → aurora/model/encoder.py b/aurora/model/perceiver_encoder.py → aurora/model/encoder.py
@@ -14,15 +14,15 @@
     pos_expansion,
     scale_expansion,
 )
-from aurora.model.groupedpatchembed import LevelPatchEmbed
 from aurora.model.helpers import (
     check_lat_lon_dtype,
     create_var_map,
     get_ids_for_var_map,
     init_weights,
 )
+from aurora.model.patchembed import LevelPatchEmbed
 from aurora.model.perceiver import MLP, PerceiverResampler
-from aurora.model.pos_encode import get_2d_patched_lat_lon_encode
+from aurora.model.posencoding import get_2d_patched_lat_lon_encode
 
 
 class Perceiver3DEncoder(nn.Module):

diff --git a/aurora/model/groupedpatchembed.py → aurora/model/patchembed.py b/aurora/model/groupedpatchembed.py → aurora/model/patchembed.py
diff --git a/aurora/model/pos_encode.py → aurora/model/posencoding.py b/aurora/model/pos_encode.py → aurora/model/posencoding.py
diff --git a/aurora/model/swin_block.py → aurora/model/swin2d.py b/aurora/model/swin_block.py → aurora/model/swin2d.py
diff --git a/aurora/model/swin_3d_block.py → aurora/model/swin3d.py b/aurora/model/swin_3d_block.py → aurora/model/swin3d.py
@@ -20,7 +20,7 @@
 from aurora.model.fourier import lead_time_expansion
 from aurora.model.helpers import Int3Tuple, init_weights, maybe_adjust_windows
 from aurora.model.lora import LoraMode
-from aurora.model.swin_block import MLP, WindowAttention, get_two_sidded_padding
+from aurora.model.swin2d import MLP, WindowAttention, get_two_sidded_padding
 
 
 def window_partition_3d(x: torch.Tensor, ws: Int3Tuple) -> torch.Tensor:

diff --git a/aurora/model/vit_block.py b/aurora/model/vit_block.py