We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b42b19f commit 2e5e353Copy full SHA for 2e5e353
torchrl/_utils.py
@@ -410,7 +410,9 @@ def accept_remote_rref_udf_invocation(decorated_class):
410
"""Class decorator that applies `accept_remote_rref_invocation` to all public methods."""
411
# ignores private methods
412
for name in dir(decorated_class):
413
- method = getattr(decorated_class, name)
+ method = getattr(decorated_class, name, None)
414
+ if method is None:
415
+ continue
416
if callable(method) and not name.startswith("_"):
417
setattr(decorated_class, name, accept_remote_rref_invocation(method))
418
return decorated_class
0 commit comments