use () key instead if class for filter

Signed-off-by: Hongpeng Guo <[email protected]>
ray-project · Oct 7, 2024 · e4d8b63 · e4d8b63
1 parent 58bbf5d
commit e4d8b63
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/ray/train/_internal/logging.py b/python/ray/train/_internal/logging.py
@@ -19,7 +19,7 @@
 RAY_TRAIN_LOG_CONFIG_PATH = os.environ.get(RAY_TRAIN_LOG_CONFIG_PATH_ENV)
 
 
-class HiddenRecordFilter:
+class HiddenRecordFilter(logging.Filter):
     """Filters out log records with the "hide" attribute set to True.
 
     This filter allows you to override default logging behavior. For example, if errors

diff --git a/python/ray/train/_internal/logging.yaml b/python/ray/train/_internal/logging.yaml
@@ -11,7 +11,7 @@ filters:
   console_filter:
     (): ray.train._internal.logging.HiddenRecordFilter
   core_context_filter:
-    class: ray._private.ray_logging.filters.CoreContextFilter
+    (): ray._private.ray_logging.filters.CoreContextFilter
 
 handlers:
   file: