Skip to content

Commit 00d3199

Browse files
author
Vincent Moens
committed
[Feature] EnvBase.check_env_specs
ghstack-source-id: 332dbf9 Pull Request resolved: #2600
1 parent de153bf commit 00d3199

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

torchrl/envs/common.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -390,6 +390,11 @@ def __init__(
390390
self.batch_size = torch.Size(batch_size)
391391
self._run_type_checks = run_type_checks
392392
self._allow_done_after_reset = allow_done_after_reset
393+
@wraps(check_env_specs_func)
394+
def check_env_specs(self, *args, **kwargs):
395+
return check_env_specs_func(self, *args, **kwargs)
396+
397+
check_env_specs.__doc__ = check_env_specs_func.__doc__
393398

394399
@classmethod
395400
def __new__(cls, *args, _inplace_update=False, _batch_locked=True, **kwargs):

0 commit comments

Comments
 (0)