[Refactor] set_default -> setdefault (#935)

tcbegley · web-flow · commit f47d2cb08955 · 2023-02-24T13:33:08.000Z
diff --git a/torchrl/envs/gym_like.py b/torchrl/envs/gym_like.py
@@ -261,7 +261,7 @@ def _reset(
             for key, item in self.observation_spec.items():
                 if key not in tensordict_out.keys():
                     tensordict_out[key] = item.zero()
-        tensordict_out.set_default(
+        tensordict_out.setdefault(
             "done",
             torch.zeros(*self.batch_size, 1, dtype=torch.bool, device=self.device),
         )

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,7 @@ def _reset(`
`261`	`261`	`for key, item in self.observation_spec.items():`
`262`	`262`	`if key not in tensordict_out.keys():`
`263`	`263`	`tensordict_out[key] = item.zero()`
`264`		`- tensordict_out.set_default(`
	`264`	`+ tensordict_out.setdefault(`
`265`	`265`	`"done",`
`266`	`266`	`torch.zeros(*self.batch_size, 1, dtype=torch.bool, device=self.device),`
`267`	`267`	`)`