From 3e20ef9fdb6fd34dfa3ca323e548134ec4967f07 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Wed, 7 Jun 2023 00:48:02 +0200 Subject: [PATCH] remove unused fixture --- tests/tests_pytorch/plugins/test_amp_plugins.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tests_pytorch/plugins/test_amp_plugins.py b/tests/tests_pytorch/plugins/test_amp_plugins.py index b9f9bd3bcb64e..fba450eccb688 100644 --- a/tests/tests_pytorch/plugins/test_amp_plugins.py +++ b/tests/tests_pytorch/plugins/test_amp_plugins.py @@ -185,7 +185,7 @@ def configure_optimizers(self): trainer.fit(model) -def test_cpu_amp_precision_context_manager(tmpdir): +def test_cpu_amp_precision_context_manager(): """Test to ensure that the context manager correctly is set to CPU + bfloat16.""" plugin = MixedPrecisionPlugin("bf16-mixed", "cpu") assert plugin.device == "cpu"