andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Improve log_metric for use in Notebooks

Browse source
This commit is contained in:
Andras Schmelczer 2022-08-07 14:13:11 +02:00
parent 58286f6f1c
commit e717234f66
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
great_ai/parameters/log_metric.py
View file

@ -5,7 +5,7 @@ from ..context import get_context
from ..tracing import TracingContext
def log_metric(argument_name: str, value: Any) -> None:
def log_metric(argument_name: str, value: Any, disable_logging: bool=False) -> None:
"""Log a key (argument_name)-value pair that is persisted inside the trace.
The name of the function from where this is called is also stored.
@ -13,12 +13,19 @@ def log_metric(argument_name: str, value: Any) -> None:
Args:
argument_name: The key for storing the value.
value: Value to log. Must be JSON-serialisable.
disable_logging: If True, only persist in trace but don't show in console
"""
tracing_context = TracingContext.get_current_tracing_context()
try:
caller = inspect.stack()[1].function
actual_name = f"metric:{caller}:{argument_name}"
except:
# inspect might not work in notebooks
actual_name = f"metric:{argument_name}"
if tracing_context:
tracing_context.log_value(name=actual_name, value=value)
if not disable_logging:
get_context().logger.info(f"{actual_name}={value}")