From dcdd8e186f83d86ef443babfff59c86e6192cdd4 Mon Sep 17 00:00:00 2001 From: Yukio Siraichi Date: Thu, 29 Feb 2024 12:20:37 -0300 Subject: [PATCH] Fix test skip. --- test/test_operations.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/test/test_operations.py b/test/test_operations.py index 84479255c481..4ee583bdc8be 100644 --- a/test/test_operations.py +++ b/test/test_operations.py @@ -67,20 +67,20 @@ def _is_on_eager_debug_mode(): 'skip on eager debug mode') -def _skipIfFunctionalization(disabled=True, reason=""): - verb = "is" if disabled else "is not" +def _skipIfFunctionalization(value=True, reason=""): + verb = "is" if value else "is not" reason = f" Reason: {reason}" if reason else "" return unittest.skipIf( - XLA_DISABLE_FUNCTIONALIZATION, + XLA_DISABLE_FUNCTIONALIZATION is value, f'Works only when functionalization {verb} disabled.{reason}.') def skipIfFunctionalizationEnabled(reason): - return _skipIfFunctionalization(disabled=False, reason=reason) + return _skipIfFunctionalization(value=False, reason=reason) def skipIfFunctionalizationDisabled(reason): - return _skipIfFunctionalization(disabled=True, reason=reason) + return _skipIfFunctionalization(value=True, reason=reason) def _gen_tensor(*args, **kwargs): @@ -994,7 +994,7 @@ def func(a, b): # TODO - upstream behavior has changed and results in expected DestroyXlaTensor # counter as of 11/13/2023. Re-enable after reviewing the change. - @skipIfFunctionalizationDisabled("metrics differ") + # @skipIfFunctionalizationDisabled("metrics differ") def test_set(self): met.clear_all()