[BugFix] Run checks when creating fake_td (#877)

vmoens · web-flow · commit c5493eccd6a4 · 2023-01-29T06:44:15.000Z
diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py
@@ -774,7 +774,7 @@ def fake_tensordict(self) -> TensorDictBase:
             },
             batch_size=self.batch_size,
             device=self.device,
-            _run_checks=False,
+            _run_checks=True,  # this method should not be run very often. This facilitates debugging
         )
         return fake_td
 

Original file line number	Diff line number	Diff line change
`@@ -774,7 +774,7 @@ def fake_tensordict(self) -> TensorDictBase:`
`774`	`774`	`},`
`775`	`775`	`batch_size=self.batch_size,`
`776`	`776`	`device=self.device,`
`777`		`- _run_checks=False,`
	`777`	`+ _run_checks=True, # this method should not be run very often. This facilitates debugging`
`778`	`778`	`)`
`779`	`779`	`return fake_td`
`780`	`780`