def test_clone(self): env = GarageEnv(DummyDiscreteEnv(obs_dim=(10, ), action_dim=4)) policy = CategoricalGRUPolicy(env_spec=env.spec) policy_clone = policy.clone('CategoricalGRUPolicyClone') assert policy.env_spec == policy_clone.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(DummyDiscreteEnv(obs_dim=(10, ), action_dim=4)) policy = CategoricalGRUPolicy(env_spec=env.spec) policy_clone = policy.clone('CategoricalGRUPolicyClone') assert policy.env_spec == policy_clone.env_spec