[BugFix] Use setdefault in _cache_values (#1910)

pytorch · Feb 15, 2024 · 45764b5 · 45764b5
1 parent 6c407bd
commit 45764b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/objectives/utils.py b/torchrl/objectives/utils.py
@@ -459,7 +459,7 @@ def _cache_values(fun):
 
     def new_fun(self, netname=None):
         __dict__ = self.__dict__
-        _cache = __dict__["_cache"]
+        _cache = __dict__.setdefault("_cache", {})
         attr_name = name
         if netname is not None:
             attr_name += "_" + netname