optimize imports

Author: HydrogenSulfate

ppsci/arch/activation.py

@@ -15,8 +15,8 @@
 from typing import Callable
 
 import paddle
-import paddle.nn as nn
 import paddle.nn.functional as F
+from paddle import nn
 
 act_func_dict = {
     "elu": F.elu,

ppsci/arch/base.py

@@ -17,7 +17,7 @@
 
 import numpy as np
 import paddle
-import paddle.nn as nn
+from paddle import nn
 
 from ppsci.utils import logger
 

ppsci/arch/embedding_koopman.py

@@ -21,7 +21,7 @@
 
 import numpy as np
 import paddle
-import paddle.nn as nn
+from paddle import nn
 from paddle.nn.initializer import Constant
 from paddle.nn.initializer import Uniform
 

ppsci/arch/mlp.py

@@ -15,7 +15,7 @@
 from typing import Tuple
 from typing import Union
 
-import paddle.nn as nn
+from paddle import nn
 
 from ppsci.arch import activation as act_mod
 from ppsci.arch import base

ppsci/data/__init__.py

@@ -17,9 +17,9 @@
 from functools import partial
 
 import numpy as np
-import paddle.device as device
 import paddle.distributed as dist
-import paddle.io as io
+from paddle import device
+from paddle import io
 
 from ppsci.data import dataloader
 from ppsci.data import dataset

ppsci/equation/pde/base.py

@@ -18,8 +18,8 @@
 from typing import Tuple
 
 import paddle
-import paddle.nn as nn
 import sympy
+from paddle import nn
 
 
 class PDE:

ppsci/loss/base.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.nn as nn
+from paddle import nn
 
 
 class Loss(nn.Layer):

ppsci/metric/base.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import paddle.nn as nn
+from paddle import nn
 
 
 class Metric(nn.Layer):

ppsci/solver/eval.py

@@ -15,7 +15,7 @@
 import time
 
 import paddle
-import paddle.io as io
+from paddle import io
 
 from ppsci.solver import printer
 from ppsci.utils import expression

ppsci/solver/solver.py

@@ -23,13 +23,13 @@
 from typing import Optional
 
 import paddle
-import paddle.amp as amp
 import paddle.distributed as dist
-import paddle.incubate as incubate
-import paddle.nn as nn
-import paddle.optimizer as optimizer
 import visualdl as vdl
 from packaging import version
+from paddle import amp
+from paddle import incubate
+from paddle import nn
+from paddle import optimizer
 from paddle.distributed import fleet
 from typing_extensions import Literal