Fix typing and minor issues

2022-07-07 14:12:44 +02:00 · 2022-07-07 14:12:44 +02:00 · 72ab627a34
commit 72ab627a34
parent 2db2253578
54 changed files with 635 additions and 589 deletions
--- a/great_ai/tracing/add_ground_truth.py
+++ b/great_ai/tracing/add_ground_truth.py
@ -51,7 +51,7 @@ def add_ground_truth(

    created = datetime.utcnow().isoformat()
    traces = [
-        Trace(
+        Trace[T](
            created=created,
            original_execution_time_ms=0,
            logged_values=X if isinstance(X, dict) else {"input": X},