fix tests

notadamking · Jul 5, 2019 · baab903 · baab903
1 parent eb54071
commit baab903
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu
@@ -1,7 +1,7 @@
 FROM python:3.6.8-jessie
 
-ADD ./requirements.txt /code/
 ADD ./requirements.base.txt /code/
+ADD ./requirements.txt /code/
 
 WORKDIR /code
 

diff --git a/lib/RLTrader.py b/lib/RLTrader.py
@@ -126,7 +126,7 @@ def optimize_params(self, trial, n_prune_evals_per_trial: int = 2, n_tests_per_e
         validation_env = SubprocVecEnv([make_env(validation_provider, i) for i in range(1)])
 
         model_params = self.optimize_agent_params(trial)
-        model = self.Model(self.Policy, train_env, verbose=self.model_verbose, nminibatches=1,
+        model = self.Model(self.PolicySub, train_env, verbose=self.model_verbose, nminibatches=1,
                            tensorboard_log=self.tensorboard_path, **model_params)
 
         last_reward = -np.finfo(np.float16).max

diff --git a/requirements.no-gpu.txt b/requirements.no-gpu.txt
@@ -1,2 +1,2 @@
--r requirements.base.yml
+-r requirements.base.txt
 tensorflow