From e8b44bbaf4db25683d701c03aee86f0ad9582a32 Mon Sep 17 00:00:00 2001 From: Trinkle23897 <463003665@qq.com> Date: Mon, 22 Jun 2020 21:39:00 +0800 Subject: [PATCH] move sac_mcc to examples (runtime too long) --- test/continuous/test_sac_with_mcc.py => examples/sac_mcc.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) rename test/continuous/test_sac_with_mcc.py => examples/sac_mcc.py (97%) diff --git a/test/continuous/test_sac_with_mcc.py b/examples/sac_mcc.py similarity index 97% rename from test/continuous/test_sac_with_mcc.py rename to examples/sac_mcc.py index 74e439a..d8b5413 100644 --- a/test/continuous/test_sac_with_mcc.py +++ b/examples/sac_mcc.py @@ -12,10 +12,7 @@ from tianshou.data import Collector, ReplayBuffer from tianshou.env import VectorEnv from tianshou.exploration import OUNoise -if __name__ == '__main__': - from net import ActorProb, Critic -else: # pytest - from test.continuous.net import ActorProb, Critic +from continuous_net import ActorProb, Critic def get_args():