move sac_mcc to examples (runtime too long)

2020-06-22 21:39:00 +08:00 · 2020-06-22 21:39:00 +08:00 · e8b44bbaf4
commit e8b44bbaf4
parent 6a2963bd64
1 changed files with 1 additions and 4 deletions
--- a/test/continuous/test_sac_with_mcc.py
+++ b/test/continuous/test_sac_with_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():