Skip to content

Commit 6cb746d

Browse files
authored
Add log_table function for mlflow (agentscope-ai#249)
1 parent 6b5edec commit 6cb746d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

trinity/utils/monitor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,8 @@ def __init__(
200200
self.console_logger = get_logger(__name__, in_ray_actor=True)
201201

202202
def log_table(self, table_name: str, experiences_table: pd.DataFrame, step: int):
203-
pass
203+
experiences_table["step"] = step
204+
mlflow.log_table(data=experiences_table, artifact_file=f"{table_name}.json")
204205

205206
def log(self, data: dict, step: int, commit: bool = False) -> None:
206207
"""Log metrics."""

0 commit comments

Comments
 (0)