diff --git a/sims/Timeloop/train_rw_timeloop.py b/sims/Timeloop/train_rw_timeloop.py index 506211d6..d9c027d2 100644 --- a/sims/Timeloop/train_rw_timeloop.py +++ b/sims/Timeloop/train_rw_timeloop.py @@ -16,7 +16,6 @@ os.sys.path.insert(0, os.path.abspath('../../')) from arch_gym.envs.TimeloopEnv import TimeloopEnv from arch_gym.envs.timeloop_acme_wrapper import make_timeloop_env -from arch_gym.envs.envHelpers import helpers from absl import app @@ -54,8 +53,6 @@ def __init__(self, script_dir, output_dir, arch_dir, mapper_dir, workload_dir, t self.config_params_obj = TimeloopConfigParams(params_file) self.arch_params = self.config_params_obj.get_all_params() - self.helper = helpers() - # Initialize timeloop wrapper self.timeloop = TimeloopWrapper()