Merge pull request #259 from mlte-team/feature/report-artifact

Report Artifact Refactor
mlte-team · Sep 20, 2023 · f8b1793 · f8b1793
2 parents 0a2f6f4 + 0a8355d
commit f8b1793
Show file tree

Hide file tree

Showing 24 changed files with 2,685 additions and 1,078 deletions.
diff --git a/mlte/artifact/model.py b/mlte/artifact/model.py
@@ -11,6 +11,7 @@
 from mlte.artifact.type import ArtifactType
 from mlte.model import BaseModel
 from mlte.negotiation.model import NegotiationCardModel
+from mlte.report.model import ReportModel
 from mlte.spec.model import SpecModel
 from mlte.validation.model import ValidatedSpecModel
 from mlte.value.model import ValueModel
@@ -36,6 +37,10 @@ class ArtifactModel(BaseModel):
     """The artifact header."""
 
     body: Union[
-        NegotiationCardModel, ValueModel, SpecModel, ValidatedSpecModel
+        NegotiationCardModel,
+        ValueModel,
+        SpecModel,
+        ValidatedSpecModel,
+        ReportModel,
     ] = Field(..., discriminator="artifact_type")
     """The artifact body."""
diff --git a/mlte/artifact/type.py b/mlte/artifact/type.py
@@ -25,3 +25,6 @@ class ArtifactType(str, Enum):
 
     VALIDATED_SPEC = auto()
     """The validated specification artifact type."""
+
+    REPORT = auto()
+    """The report artifact type."""