From e335f05fb15fec92b523f28fc4d9f019a35b7e75 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Tue, 13 Sep 2022 15:03:36 +0200 Subject: [PATCH] Rename test_scheduler_outputs_equivalence in model tests. (#451) --- tests/test_modeling_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 7c098adb..56356461 100644 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -179,7 +179,7 @@ class ModelTesterMixin: loss.backward() ema_model.step(model) - def test_scheduler_outputs_equivalence(self): + def test_outputs_equivalence(self): def set_nan_tensor_to_zero(t): # Temporary fallback until `aten::_index_put_impl_` is implemented in mps # Track progress in https://github.com/pytorch/pytorch/issues/77764