Limit warning filter to torchrl (#2762)

pytorch · Feb 7, 2025 · 85d1e70 · 85d1e70
1 parent 3be85c6
commit 85d1e70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/_utils.py b/torchrl/_utils.py
@@ -67,7 +67,7 @@
 _os_is_windows = sys.platform == "win32"
 RL_WARNINGS = strtobool(os.environ.get("RL_WARNINGS", "1"))
 if RL_WARNINGS:
-    warnings.simplefilter("once", DeprecationWarning)
+    warnings.filterwarnings("once", category=DeprecationWarning, module="torchrl")
 
 BATCHED_PIPE_TIMEOUT = float(os.environ.get("BATCHED_PIPE_TIMEOUT", "10000.0"))