Rename test_scheduler_outputs_equivalence in model tests. (#451)

e335f05f · Pedro Cuenca · GitHub · f7cd6b87 · e335f05f
Unverified Commit e335f05f authored Sep 13, 2022 by Pedro Cuenca Committed by GitHub Sep 13, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_common.py tests/test_modeling_common.py +1 -1

No files found.
--- 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