Skip to content

Commit

Permalink
additional logging
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosgjs committed Jan 19, 2024
1 parent f691fff commit 4de3052
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/autora/doc/pipelines/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,9 @@ def eval(
logger.info(f"running predict with {data_file}")
logger.info(f"model path: {model_path}")
mlflow.log_params(param_dict)
mlflow.log_param("prompt_id", prompt_id)
mlflow.log_param("model_path", model_path)
mlflow.log_param("data_file", data_file)

with jsonlines.open(data_file) as reader:
items = [item for item in reader]
Expand Down

0 comments on commit 4de3052

Please sign in to comment.