Skip to content

Commit 0e093c5

Browse files
add import PrecipNet in ppsci/arch/__init__.py
1 parent 5e01942 commit 0e093c5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ppsci/arch/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from ppsci.arch.physx_transformer import PhysformerGPT2 # isort:skip
2222
from ppsci.arch.model_list import ModelList # isort:skip
2323
from ppsci.arch.afno import AFNONet # isort:skip
24+
from ppsci.arch.afno import PrecipNet # isort:skip
2425
from ppsci.utils import logger # isort:skip
2526

2627

@@ -32,6 +33,7 @@
3233
"PhysformerGPT2",
3334
"ModelList",
3435
"AFNONet",
36+
"PrecipNet",
3537
"build_model",
3638
]
3739

0 commit comments

Comments
 (0)