From 5fe92613d2a4c67db5055859f7eef9aac53962d3 Mon Sep 17 00:00:00 2001 From: Aliaksandr Kuzmik Date: Tue, 26 Nov 2024 17:17:13 +0100 Subject: [PATCH] Add new log message for experiment item upload --- sdks/python/src/opik/evaluation/scorer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sdks/python/src/opik/evaluation/scorer.py b/sdks/python/src/opik/evaluation/scorer.py index 80237371ba..7d9e742d1b 100644 --- a/sdks/python/src/opik/evaluation/scorer.py +++ b/sdks/python/src/opik/evaluation/scorer.py @@ -1,5 +1,6 @@ import tqdm import logging +import time from concurrent import futures from typing import List, Optional, Dict, Any, Union, Callable @@ -114,8 +115,10 @@ def _process_item( dataset_item_id=item.id, trace_id=trace_data.id, ) - + insert_start = time.time() + LOGGER.debug("Experiment item insertion started: %s", experiment_item_.trace_id) experiment_.insert(experiment_items=[experiment_item_]) + LOGGER.debug("Experiment item insertion ended : %s, with time %s", experiment_item_.trace_id, time.time() - insert_start) def score_tasks(