From 3d96b4e36c479ff81eb35ba78fd2d8c5584ea7ae Mon Sep 17 00:00:00 2001
From: Peter Webb <peter.webb@dbtlabs.com>
Date: Mon, 21 Oct 2024 19:01:15 -0400
Subject: [PATCH] Loosen Type in TimingInfo (#10897)

---
 core/dbt/artifacts/schemas/results.py | 8 +++-----
 schemas/dbt/run-results/v6.json       | 6 +-----
 schemas/dbt/sources/v3.json           | 6 +-----
 3 files changed, 5 insertions(+), 15 deletions(-)

diff --git a/core/dbt/artifacts/schemas/results.py b/core/dbt/artifacts/schemas/results.py
index ee27fc6d5d4..dd455f309b8 100644
--- a/core/dbt/artifacts/schemas/results.py
+++ b/core/dbt/artifacts/schemas/results.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Union
+from typing import Any, Callable, Dict, List, Optional, Sequence, Union
 
 from dbt.contracts.graph.nodes import ResultNode
 from dbt_common.dataclass_schema import StrEnum, dbtClassMixin
@@ -16,7 +16,7 @@ class TimingInfo(dbtClassMixin):
     Do not call directly, use `collect_timing_info` instead.
     """
 
-    name: Literal["compile", "execute", "other"]
+    name: str
     started_at: Optional[datetime] = None
     completed_at: Optional[datetime] = None
 
@@ -37,9 +37,7 @@ def to_msg_dict(self):
 
 # This is a context manager
 class collect_timing_info:
-    def __init__(
-        self, name: Literal["compile", "execute", "other"], callback: Callable[[TimingInfo], None]
-    ) -> None:
+    def __init__(self, name: str, callback: Callable[[TimingInfo], None]) -> None:
         self.timing_info = TimingInfo(name=name)
         self.callback = callback
 
diff --git a/schemas/dbt/run-results/v6.json b/schemas/dbt/run-results/v6.json
index 96456882ae6..1bf1cf75e83 100644
--- a/schemas/dbt/run-results/v6.json
+++ b/schemas/dbt/run-results/v6.json
@@ -84,11 +84,7 @@
               "title": "TimingInfo",
               "properties": {
                 "name": {
-                  "enum": [
-                    "compile",
-                    "execute",
-                    "other"
-                  ]
+                  "type": "string"
                 },
                 "started_at": {
                   "anyOf": [
diff --git a/schemas/dbt/sources/v3.json b/schemas/dbt/sources/v3.json
index 8cb3633f99a..df2784f1a81 100644
--- a/schemas/dbt/sources/v3.json
+++ b/schemas/dbt/sources/v3.json
@@ -211,11 +211,7 @@
                   "title": "TimingInfo",
                   "properties": {
                     "name": {
-                      "enum": [
-                        "compile",
-                        "execute",
-                        "other"
-                      ]
+                      "type": "string"
                     },
                     "started_at": {
                       "anyOf": [