Refine lib -> xautodl

tests/test_math_adv.py

@@ -4,14 +4,12 @@
 # pytest tests/test_math_adv.py -s                  #
 #####################################################
 import unittest
-import pytest
 
-from xautodl import datasets
-from datasets.math_core import QuadraticFunc
-from datasets.math_core import ConstantFunc
-from datasets.math_core import DynamicLinearFunc
-from datasets.math_core import DynamicQuadraticFunc
-from datasets.math_core import ComposedSinFunc
+from xautodl.datasets.math_core import QuadraticFunc
+from xautodl.datasets.math_core import ConstantFunc
+from xautodl.datasets.math_core import DynamicLinearFunc
+from xautodl.datasets.math_core import DynamicQuadraticFunc
+from xautodl.datasets.math_core import ComposedSinFunc
 
 
 class TestConstantFunc(unittest.TestCase):

xautodl/models/shape_infers/InferCifarResNet.py

@@ -3,6 +3,7 @@
 #####################################################
 import torch.nn as nn
 import torch.nn.functional as F
+
 from ..initialization import initialize_resnet
 
 

xautodl/models/shape_infers/InferMobileNetV2.py

@@ -2,7 +2,9 @@
 # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019.01 #
 #####################################################
 # MobileNetV2: Inverted Residuals and Linear Bottlenecks, CVPR 2018
+#####################################################
 from torch import nn
+
 from ..initialization import initialize_resnet
 from ..SharedUtils import parse_channel_info
 

xautodl/models/shape_infers/InferTinyCellNet.py

@@ -3,8 +3,9 @@
 #####################################################
 from typing import List, Text, Any
 import torch.nn as nn
-from models.cell_operations import ResNetBasicblock
-from models.cell_infers.cells import InferCell
+
+from ..cell_operations import ResNetBasicblock
+from ..cell_infers.cells import InferCell
 
 
 class DynamicShapeTinyNet(nn.Module):