Fix an error in setup.py

iffiX · Mar 26, 2021 · ac37b22 · ac37b22
1 parent d0d6faf
commit ac37b22
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 2 deletions.
diff --git a/machin/__main__.py b/machin/__main__.py
diff --git a/machin/frame/auto/launcher.py b/machin/frame/auto/launcher.py
@@ -5,7 +5,12 @@
 from .config import init_algorithm_from_config, is_algorithm_distributed
 from .pl_plugin import DDPPlugin, DDPSpawnPlugin
 from .pl_logger import LocalMediaLogger
-from .dataset import *
+from .dataset import (
+    is_discrete_space,
+    is_continuous_space,
+    RLGymDiscActDataset,
+    RLGymContActDataset
+)
 
 
 class Launcher(pl.LightningModule):

diff --git a/setup.py b/setup.py
@@ -53,7 +53,7 @@
         "psutil",
         "numpy",
         "torch>=1.6.0",
-        "pytorch-lightning>=1.0"
+        "pytorch-lightning>=1.0",
         "torchviz",
         "moviepy",
         "matplotlib",

diff --git a/test/frame/auto/test_dataset.py b/test/frame/auto/test_dataset.py
diff --git a/test/frame/auto/test_pl_logger.py b/test/frame/auto/test_pl_logger.py
diff --git a/test/frame/auto/test_pl_plugin.py b/test/frame/auto/test_pl_plugin.py