adjusted tests

JuliaReinforcementLearning · Jul 19, 2023 · 309fbf5 · 309fbf5
1 parent 82f77f4
commit 309fbf5
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 91 deletions.
diff --git a/src/ReinforcementLearningExperiments/deps/experiments/experiments/MARL/DQN_mpe_simple.jl b/src/ReinforcementLearningExperiments/deps/experiments/experiments/MARL/DQN_mpe_simple.jl
@@ -24,7 +24,7 @@ function RLCore.Experiment(
     is_enable_double_DQN=true
 )
     rng = StableRNG(seed)
-    env = discrete2standard_discrete(PettingzooEnv("mpe.simple_v2"; seed=seed))
+    env = discrete2standard_discrete(PettingZooEnv("mpe.simple_v2"; seed=seed))
     ns, na = length(state(env)), length(action_space(env))
 
     agent = Agent(

diff --git a/src/ReinforcementLearningExperiments/src/ReinforcementLearningExperiments.jl b/src/ReinforcementLearningExperiments/src/ReinforcementLearningExperiments.jl
@@ -1,6 +1,7 @@
 module ReinforcementLearningExperiments
 
 using Reexport
+using Requires
 
 @reexport using ReinforcementLearningCore, ReinforcementLearningBase, ReinforcementLearningZoo
 
@@ -20,6 +21,9 @@ include(joinpath(EXPERIMENTS_DIR, "JuliaRL_VPG_CartPole.jl"))
 include(joinpath(EXPERIMENTS_DIR, "JuliaRL_TRPO_CartPole.jl"))
 include(joinpath(EXPERIMENTS_DIR, "JuliaRL_MPO_CartPole.jl"))
 include(joinpath(EXPERIMENTS_DIR, "IDQN_TicTacToe.jl"))
+@require PyCall = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0" include(
+    joinpath(EXPERIMENTS_DIR, "DQN_mpe_simple.jl")
+)
 
 # dynamic loading environments
 function __init__() end

diff --git a/src/ReinforcementLearningExperiments/test/experiments/DQN_mpe_simple.jl b/src/ReinforcementLearningExperiments/test/experiments/DQN_mpe_simple.jl
diff --git a/src/ReinforcementLearningExperiments/test/runtests.jl b/src/ReinforcementLearningExperiments/test/runtests.jl
@@ -3,10 +3,6 @@ using CUDA
 
 using Requires
 
-const EXPERIMENTS_DIR = joinpath(@__DIR__, "experiments")
-@require PyCall = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0" include(
-    joinpath(EXPERIMENTS_DIR, "DQN_mpe_simple.jl")
-)
 
 
 CUDA.allowscalar(false)