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

Fix function

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-08 20:33:33 +02:00
parent 299d644bc4
commit 7c99305afb
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
1 changed files with 15 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

9
great_ai/tracing/add_ground_truth.py
View file

@ -1,7 +1,8 @@
from datetime import datetime from datetime import datetime
from math import ceil from math import ceil
from random import shuffle 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 ( from ..constants import (
GROUND_TRUTH_TAG_NAME, GROUND_TRUTH_TAG_NAME,
@ -51,7 +52,10 @@ def add_ground_truth(
created = datetime.utcnow().isoformat() created = datetime.utcnow().isoformat()
traces = [ traces = [
Trace[T]( cast(
Trace[T],
Trace( # avoid ValueError: "Trace" object has no field "__orig_class__"
trace_id=str(uuid4()),
created=created, created=created,
original_execution_time_ms=0, original_execution_time_ms=0,
logged_values=X if isinstance(X, dict) else {"input": X}, logged_values=X if isinstance(X, dict) else {"input": X},
@ -60,6 +64,7 @@ def add_ground_truth(
feedback=y, feedback=y,
exception=None, exception=None,
tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags], tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags],
),
) )
for ((X, y), split_tag) in zip(values, split_tags) for ((X, y), split_tag) in zip(values, split_tags)
] ]