Fix Ruff rule LOG

tensorflow · Oct 27, 2024 · bf43171 · bf43171
1 parent 5c2bdfd
commit bf43171
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tfx/utils/logging_utils_test.py b/tfx/utils/logging_utils_test.py
@@ -49,9 +49,9 @@ def testDefaultSettings(self):
 
   def testOverrideSettings(self):
     """Ensure log overrides are set correctly."""
-    config = logging_utils.LoggerConfig(log_root='path', log_level=logging.WARN,
+    config = logging_utils.LoggerConfig(log_root='path', log_level=logging.WARNING,
                                         pipeline_name='pipe', worker_name='wrk')
     self.assertEqual(config.log_root, 'path')
-    self.assertEqual(config.log_level, logging.WARN)
+    self.assertEqual(config.log_level, logging.WARNING)
     self.assertEqual(config.pipeline_name, 'pipe')
     self.assertEqual(config.worker_name, 'wrk')