From 7c99305afb4f1463cfe2953c34a445cb984b8ea0 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Fri, 8 Jul 2022 20:33:33 +0200 Subject: [PATCH] Fix function --- great_ai/tracing/add_ground_truth.py | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/great_ai/tracing/add_ground_truth.py b/great_ai/tracing/add_ground_truth.py index 78cc7cc..e6e7982 100644 --- a/great_ai/tracing/add_ground_truth.py +++ b/great_ai/tracing/add_ground_truth.py @@ -1,7 +1,8 @@ from datetime import datetime from math import ceil from random import shuffle -from typing import Any, Iterable, List, TypeVar +from typing import Any, Iterable, List, TypeVar, cast +from uuid import uuid4 from ..constants import ( GROUND_TRUTH_TAG_NAME, @@ -51,15 +52,19 @@ def add_ground_truth( created = datetime.utcnow().isoformat() traces = [ - Trace[T]( - created=created, - original_execution_time_ms=0, - logged_values=X if isinstance(X, dict) else {"input": X}, - models=[], - output=y, - feedback=y, - exception=None, - tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags], + cast( + Trace[T], + Trace( # avoid ValueError: "Trace" object has no field "__orig_class__" + trace_id=str(uuid4()), + created=created, + original_execution_time_ms=0, + logged_values=X if isinstance(X, dict) else {"input": X}, + models=[], + output=y, + feedback=y, + exception=None, + tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags], + ), ) for ((X, y), split_tag) in zip(values, split_tags) ]