Fix function

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](
+        cast(
-            created=created,
+            Trace[T],
-            original_execution_time_ms=0,
+            Trace(  # avoid ValueError: "Trace" object has no field "__orig_class__"
-            logged_values=X if isinstance(X, dict) else {"input": X},
+                trace_id=str(uuid4()),
-            models=[],
+                created=created,
-            output=y,
+                original_execution_time_ms=0,
-            feedback=y,
+                logged_values=X if isinstance(X, dict) else {"input": X},
-            exception=None,
+                models=[],
-            tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags],
+                output=y,
+                feedback=y,
+                exception=None,
+                tags=[GROUND_TRUTH_TAG_NAME, split_tag, *tags],
+            ),
         )
         for ((X, y), split_tag) in zip(values, split_tags)
     ]