[Quality] Make sure deprec warnings are displayed (#2088)

pytorch · Apr 18, 2024 · ee8cafb · ee8cafb
1 parent fc8ccd9
commit ee8cafb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torchrl/_utils.py b/torchrl/_utils.py
@@ -48,6 +48,9 @@
 VERBOSE = strtobool(os.environ.get("VERBOSE", "0"))
 _os_is_windows = sys.platform == "win32"
 RL_WARNINGS = strtobool(os.environ.get("RL_WARNINGS", "1"))
+if RL_WARNINGS:
+    warnings.simplefilter("once", DeprecationWarning)
+
 BATCHED_PIPE_TIMEOUT = float(os.environ.get("BATCHED_PIPE_TIMEOUT", "10000.0"))