Update message

Lightning-AI · Jul 28, 2020 · 7013897 · 7013897
1 parent 82d853b
commit 7013897
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/pytorch_lightning/trainer/ignored_warnings.py b/pytorch_lightning/trainer/ignored_warnings.py
@@ -3,12 +3,9 @@
 
 def ignore_scalar_return_in_dp():
     # Users get confused by this warning so we silence it
-    m_1 = (
-        'Was asked to gather along dimension 0, but all '
-        'input tensors were scalars; will instead unsqueeze '
-        'and return a vector.'
-    )
-    warnings.filterwarnings('ignore', message=m_1)
+    warnings.filterwarnings('ignore', message='Was asked to gather along dimension 0, but all '
+                                              'input tensors were scalars; will instead unsqueeze '
+                                              'and return a vector.')
 
 
 ignore_scalar_return_in_dp()