def test_clone(self): env = GymEnv(DummyBoxEnv(obs_dim=(4, ), action_dim=(4, ))) policy = GaussianGRUPolicy(env_spec=env.spec) policy_clone = policy.clone('GaussianGRUPolicyClone') assert policy_clone.env_spec == policy.env_spec for cloned_param, param in zip(policy_clone.parameters.values(), policy.parameters.values()): assert np.array_equal(cloned_param, param)
def test_clone(self): env = GarageEnv(DummyBoxEnv(obs_dim=(4, ), action_dim=(4, ))) policy = GaussianGRUPolicy(env_spec=env.spec) policy_clone = policy.clone('GaussianGRUPolicyClone') assert policy_clone.env_spec == policy.env_spec