pyglove 0.4.5.dev202504260808__py3-none-any.whl → 0.4.5.dev202504270808__py3-none-any.whl

pyglove/core/logging.py

@@ -17,17 +17,47 @@ This module allows PyGlove to use external created logger for logging PyGlove
 events without introducing library dependencies in PyGlove.
 """
 
+import inspect
 import logging
+from typing import Any, Callable, List, Union
 
 
 _DEFAULT_LOGGER = logging.getLogger()
 
 
+def register_frame_to_skip(
+    method: Union[Callable[..., Any], List[Callable[..., Any]]]
+) -> bool:
+  """Skips the source of the given method when logging.
+
+  Args:
+    method: The method to skip. Can be a single method or a list of methods.
+
+  Returns:
+    True if the method is registered to skip.
+
+  Raises:
+    TypeError: The source file of the method cannot be inspected.
+  """
+  register_fn = getattr(
+      _DEFAULT_LOGGER.__class__, 'register_frame_to_skip', None
+  )
+  if register_fn is None:
+    return False
+  methods = [method] if not isinstance(method, list) else method
+  for m in methods:
+    register_fn(inspect.getsourcefile(m), m.__name__)
+  return True
+
+
 def set_logger(logger: logging.Logger) -> None:
   """Sets current logger."""
   global _DEFAULT_LOGGER
   _DEFAULT_LOGGER = logger
 
+  # Skip logging frames in pyglove.logging.
+  register_frame_to_skip([debug, info, warning, error, critical])
+
 
 def get_logger() -> logging.Logger:
   """Gets the current logger."""

{pyglove-0.4.5.dev202504260808.dist-info → pyglove-0.4.5.dev202504270808.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyglove
-Version: 0.4.5.dev202504260808
+Version: 0.4.5.dev202504270808
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors

{pyglove-0.4.5.dev202504260808.dist-info → pyglove-0.4.5.dev202504270808.dist-info}/RECORD

@@ -1,6 +1,6 @@
 pyglove/__init__.py,sha256=YPUWALRDu4QuI7N3TUmVaVr513hUt0qXrjY582I1q5c,1352
 pyglove/core/__init__.py,sha256=BF-8v1D5lTBAo4vlpDt0qyOG9NQ71oh_UfthBVrPaN4,9614
-pyglove/core/logging.py,sha256=Sr5nLyUQmc4CAEYAq8qbP3ghpjmpz2sOuhq2A0tgQ6I,2456
+pyglove/core/logging.py,sha256=pD5A6K8TOWoQdYZ1IgGHHd_vIcqtcWQ51Eep5Hswfwc,3305
 pyglove/core/logging_test.py,sha256=3z_c6wnxbqDbwUmSOAZzeDPXvzoweYL5QHUQVMJ5Xgo,1917
 pyglove/core/coding/__init__.py,sha256=tuHIg19ZchtkOQbdFVTVLkUpBa5f1eo66XtnKw3lcIU,1645
 pyglove/core/coding/errors.py,sha256=aP3Y4amBzOKdlb5JnESJ3kdoijQXbiBiPDMeA88LNrk,3310
@@ -212,8 +212,8 @@ pyglove/ext/scalars/randoms.py,sha256=LkMIIx7lOq_lvJvVS3BrgWGuWl7Pi91-lA-O8x_gZs
 pyglove/ext/scalars/randoms_test.py,sha256=nEhiqarg8l_5EOucp59CYrpO2uKxS1pe0hmBdZUzRNM,2000
 pyglove/ext/scalars/step_wise.py,sha256=IDw3tuTpv0KVh7AN44W43zqm1-E0HWPUlytWOQC9w3Y,3789
 pyglove/ext/scalars/step_wise_test.py,sha256=TL1vJ19xVx2t5HKuyIzGoogF7N3Rm8YhLE6JF7i0iy8,2540
-pyglove-0.4.5.dev202504260808.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.4.5.dev202504260808.dist-info/METADATA,sha256=cO3gt2hth4HqVh7ao1AzuahePKJK_UbC1tIf60FtE_k,7089
-pyglove-0.4.5.dev202504260808.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-pyglove-0.4.5.dev202504260808.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.4.5.dev202504260808.dist-info/RECORD,,
+pyglove-0.4.5.dev202504270808.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.4.5.dev202504270808.dist-info/METADATA,sha256=uUnAmPm01eYJ9lrXzgb0ctaoqxr3UWbTCFKhgdi-oJU,7089
+pyglove-0.4.5.dev202504270808.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+pyglove-0.4.5.dev202504270808.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.4.5.dev202504270808.dist-info/RECORD,,