bugfix

test.py

@@ -187,6 +187,14 @@ class MineRLSQILBaselineAgent(MineRLAgentBase):
         self.env = env
 
     def load_agent(self):
+        """
+        boundaries = cached_reward_boundary(
+            cache_dir=os.environ.get('BOUNDARY_CACHE'),
+            env_id=MINERL_GYM_ENV,
+            n_groups=OPTION_N_GROUPS,
+            random_state=KMEANS_SEED)
+        reward_channel_scale = 1. / boundaries[-1]
+        """
         self.agent = get_agent(
             n_actions=self.env.action_space.n, arch=ARCH, n_input_channels=self.env.observation_space.shape[0],
             final_epsilon=FINAL_EPSILON,
@@ -238,12 +246,6 @@ def main():
         random_state=KMEANS_SEED,
         subtask_reward_max=SUBTASK_REWARD_MAX,
         only_vector_converter=True)
-    boundaries = cached_reward_boundary(
-        cache_dir=os.environ.get('BOUNDARY_CACHE'),
-        env_id=args.env,
-        n_groups=args.option_n_groups,
-        random_state=args.seed)
-    reward_channel_scale = 1. / boundaries[-1]
 
     def wrapper(env):
         return wrap_env(