remove DeprecationWarning class to make the warning show up in sterr.

Lightning-AI · Jun 16, 2021 · 5814213 · 5814213
1 parent 8092946
commit 5814213
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/connectors/logger_connector/result.py b/pytorch_lightning/trainer/connectors/logger_connector/result.py
@@ -286,7 +286,7 @@ def check_fn(v):
                 warning_cache.warn(
                     f"One of the returned values {set(extra.keys())} has a `grad_fn`. We will detach it automatically"
                     " but this behaviour will change in v1.6. Please detach it manually:"
-                    " `return {'loss': ..., 'something': something.detach()}`", DeprecationWarning
+                    " `return {'loss': ..., 'something': something.detach()}`",
                 )
                 return v.detach()
             return v