Refine lib -> xautodl

tests/test_super_transformer.py

@@ -3,19 +3,11 @@
 #####################################################
 # pytest ./tests/test_super_model.py -s             #
 #####################################################
-import sys, random
 import unittest
-import pytest
-from pathlib import Path
-
-lib_dir = (Path(__file__).parent / ".." / "lib").resolve()
-print("library path: {:}".format(lib_dir))
-if str(lib_dir) not in sys.path:
-    sys.path.insert(0, str(lib_dir))
 
 import torch
-from xlayers.super_core import SuperRunMode
-from trade_models import get_transformer
+from xautodl.xlayers.super_core import SuperRunMode
+from xautodl.trade_models import get_transformer
 
 
 class TestSuperTransformer(unittest.TestCase):

tests/test_synthetic_env.py

@@ -3,18 +3,10 @@
 #####################################################
 # pytest tests/test_synthetic_env.py -s             #
 #####################################################
-import sys, random
 import unittest
-import pytest
-from pathlib import Path
 
-lib_dir = (Path(__file__).parent / ".." / "lib").resolve()
-print("library path: {:}".format(lib_dir))
-if str(lib_dir) not in sys.path:
-    sys.path.insert(0, str(lib_dir))
-
-from datasets.math_core import ConstantFunc, ComposedSinFunc
-from datasets.synthetic_core import SyntheticDEnv
+from xautodl.datasets.math_core import ConstantFunc, ComposedSinFunc
+from xautodl.datasets.synthetic_core import SyntheticDEnv
 
 
 class TestSynethicEnv(unittest.TestCase):

tests/test_synthetic_utils.py

@@ -3,17 +3,9 @@
 #####################################################
 # pytest tests/test_synthetic_utils.py -s           #
 #####################################################
-import sys, random
 import unittest
-import pytest
-from pathlib import Path
 
-lib_dir = (Path(__file__).parent / ".." / "lib").resolve()
-print("library path: {:}".format(lib_dir))
-if str(lib_dir) not in sys.path:
-    sys.path.insert(0, str(lib_dir))
-
-from datasets.synthetic_core import TimeStamp
+from xautodl.datasets.synthetic_core import TimeStamp
 
 
 class TestTimeStamp(unittest.TestCase):