Skip to content

Commit 7013897

Browse files
committed
Update message
1 parent 82d853b commit 7013897

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

pytorch_lightning/trainer/ignored_warnings.py

+3-6
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,9 @@
33

44
def ignore_scalar_return_in_dp():
55
# Users get confused by this warning so we silence it
6-
m_1 = (
7-
'Was asked to gather along dimension 0, but all '
8-
'input tensors were scalars; will instead unsqueeze '
9-
'and return a vector.'
10-
)
11-
warnings.filterwarnings('ignore', message=m_1)
6+
warnings.filterwarnings('ignore', message='Was asked to gather along dimension 0, but all '
7+
'input tensors were scalars; will instead unsqueeze '
8+
'and return a vector.')
129

1310

1411
ignore_scalar_return_in_dp()

0 commit comments

Comments
 (0)