[Minor] Fix bad name in pytree warning

Vincent Moens · Vincent Moens · commit 8e3fb981c4d3 · 2025-05-16T12:37:33.000+01:00
ghstack-source-id: 873fb75 Pull-Request-resolved: #1322 (cherry picked from commit ed1f3b4)
diff --git a/tensordict/nn/cudagraphs.py b/tensordict/nn/cudagraphs.py
@@ -211,7 +211,7 @@ def __init__(
                 if not strtobool(os.environ.get("EXCLUDE_TD_FROM_PYTREE", "0")):
                     warnings.warn(
                         f"Tensordict is registered in PyTree. This is incompatible with {self.__class__.__name__}. "
-                        f"Removing TDs from PyTree. To silence this warning, call tensordict.nn.functional_module._exclude_td_from_pytree().set() "
+                        f"Removing TDs from PyTree. To silence this warning, call tensordict.nn.functional_modules._exclude_td_from_pytree().set() "
                         f"or set the environment variable `EXCLUDE_TD_FROM_PYTREE=1`. "
                         f"This operation is irreversible."
                     )

Original file line number	Diff line number	Diff line change
`@@ -211,7 +211,7 @@ def __init__(`
`211`	`211`	`if not strtobool(os.environ.get("EXCLUDE_TD_FROM_PYTREE", "0")):`
`212`	`212`	`warnings.warn(`
`213`	`213`	`f"Tensordict is registered in PyTree. This is incompatible with {self.__class__.__name__}. "`
`214`		`- f"Removing TDs from PyTree. To silence this warning, call tensordict.nn.functional_module._exclude_td_from_pytree().set() "`
	`214`	`+ f"Removing TDs from PyTree. To silence this warning, call tensordict.nn.functional_modules._exclude_td_from_pytree().set() "`
`215`	`215`	f"or set the environment variable `EXCLUDE_TD_FROM_PYTREE=1`. "
`216`	`216`	`f"This operation is irreversible."`
`217`	`217`	`)`