Skip to content

Commit 32e1dce

Browse files
authored
Update activations path in dlwp tests (#1217)
* Update activations path in dlwp tests * Update example paths
1 parent 0012209 commit 32e1dce

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

examples/weather/dlwp_healpix/configs/model/modules/activations/capped_gelu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,5 @@
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
1616

17-
_target_: physicsnemo.models.layers.activations.CappedGELU
17+
_target_: physicsnemo.nn.activations.CappedGELU
1818
cap_value: 10

examples/weather/dlwp_healpix/configs/model/modules/activations/capped_leaky_relu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,5 @@
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
1616

17-
_target_: physicsnemo.models.layers.activations.CappedLeakyReLU
17+
_target_: physicsnemo.nn.activations.CappedLeakyReLU
1818
cap_value: 10

test/models/dlwp_healpix/test_healpix_recunet_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
@pytest.fixture
3030
def conv_next_block_dict(in_channels=3, out_channels=1):
3131
activation_block = {
32-
"_target_": "physicsnemo.models.layers.activations.CappedGELU",
32+
"_target_": "physicsnemo.nn.activations.CappedGELU",
3333
"cap_value": 10,
3434
}
3535
conv_block = {
@@ -72,7 +72,7 @@ def encoder_dict(conv_next_block_dict, down_sampling_block_dict, recurrent_block
7272
def up_sampling_block_dict(in_channels=3, out_channels=1):
7373
"""Block dict fixture."""
7474
activation_block = {
75-
"_target_": "physicsnemo.models.layers.activations.CappedGELU",
75+
"_target_": "physicsnemo.nn.activations.CappedGELU",
7676
"cap_value": 10,
7777
}
7878
up_sampling_block = {

test/models/dlwp_healpix/test_healpix_unet_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
@pytest.fixture
3030
def conv_next_block_dict(in_channels=3, out_channels=1):
3131
activation_block = {
32-
"_target_": "physicsnemo.models.layers.activations.CappedGELU",
32+
"_target_": "physicsnemo.nn.activations.CappedGELU",
3333
"cap_value": 10,
3434
}
3535
conv_block = {
@@ -58,7 +58,7 @@ def down_sampling_block_dict():
5858
def up_sampling_block_dict(in_channels=3, out_channels=1):
5959
"""Upsampling dict fixture."""
6060
activation_block = {
61-
"_target_": "physicsnemo.models.layers.activations.CappedGELU",
61+
"_target_": "physicsnemo.nn.activations.CappedGELU",
6262
"cap_value": 10,
6363
}
6464
up_sampling_block = {

0 commit comments

Comments
 (0)