Skip to content

Commit d80ca08

Browse files
committed
update
1 parent 17c92fd commit d80ca08

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/tests_fabric/plugins/precision/test_fsdp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def test_fsdp_precision_config(precision, expected):
4444

4545
warning_ctx = pytest.warns if precision in ("16-true", "bf16-true") else null_ctx
4646

47-
with warning_ctx(UserWarning, match="enables mixed-precision execution"):
47+
with warning_ctx(UserWarning, match="enables computation in lower precision"):
4848
config = plugin.mixed_precision_config
4949

5050
assert config.param_dtype == expected[0]

tests/tests_pytorch/plugins/precision/test_fsdp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def test_fsdp_precision_config(precision, expected):
4444

4545
warning_ctx = pytest.warns if precision in ("16-true", "bf16-true") else null_ctx
4646

47-
with warning_ctx(UserWarning, match="enables mixed-precision execution"):
47+
with warning_ctx(UserWarning, match="enables computation in lower precision"):
4848
config = plugin.mixed_precision_config
4949

5050
assert config.param_dtype == expected[0]

0 commit comments

Comments
 (0)