[Quality] Warn if the sampler is not prioritized but update_priority …

…is called (pytorch#2253)
ai4co · Jun 27, 2024 · 1083b35 · 1083b35
1 parent 353c897
commit 1083b35
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torchrl/data/replay_buffers/samplers.py b/torchrl/data/replay_buffers/samplers.py
@@ -63,6 +63,9 @@ def update_priority(
         *,
         storage: Storage | None = None,
     ) -> dict | None:
+        warnings.warn(
+            f"Calling update_priority() on a sampler {type(self).__name__} that is not prioritized. Make sure this is the indented behaviour."
+        )
         return
 
     def mark_update(