feat: adding create field to QCEvaluation

AllenNeuralDynamics · Dec 11, 2024 · 524295c · 524295c
1 parent 9f70899
commit 524295c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/examples/quality_control.json b/examples/quality_control.json
@@ -83,7 +83,8 @@
          "tags": null,
          "notes": "",
          "allow_failed_metrics": false,
-         "latest_status": "Pending"
+         "latest_status": "Pending",
+         "created": "2022-11-22T00:00:00Z"
       },
       {
          "modality": {
@@ -126,7 +127,8 @@
          "tags": null,
          "notes": "Pass when video_1_num_frames==video_2_num_frames",
          "allow_failed_metrics": false,
-         "latest_status": "Pass"
+         "latest_status": "Pass",
+         "created": "2022-11-22T00:00:00Z"
       },
       {
          "modality": {
@@ -183,7 +185,8 @@
          "tags": null,
          "notes": null,
          "allow_failed_metrics": false,
-         "latest_status": "Pass"
+         "latest_status": "Pass",
+         "created": "2022-11-22T00:00:00Z"
       }
    ],
    "notes": null

diff --git a/examples/quality_control.py b/examples/quality_control.py
@@ -52,6 +52,7 @@
         QCMetric(name="Probe C drift", value="Low", reference="ecephys-drift-map", status_history=[s]),
     ],
     notes="",
+    created=t,
 )
 
 eval1 = QCEvaluation(
@@ -63,6 +64,7 @@
         QCMetric(name="video_2_num_frames", value=662, status_history=[s]),
     ],
     notes="Pass when video_1_num_frames==video_2_num_frames",
+    created=t,
 )
 
 eval2 = QCEvaluation(
@@ -74,6 +76,7 @@
         QCMetric(name="ProbeB_success", value=True, status_history=[s]),
         QCMetric(name="ProbeC_success", value=True, status_history=[s]),
     ],
+    created=t,
 )
 
 q = QualityControl(evaluations=[eval0, eval1, eval2])

diff --git a/src/aind_data_schema/core/quality_control.py b/src/aind_data_schema/core/quality_control.py
@@ -93,6 +93,7 @@ class QCEvaluation(AindModel):
         ),
     )
     latest_status: Status = Field(default=None, title="Evaluation status")
+    created: AwareDatetimeWithDefault = Field(default_factory=lambda: datetime.now(tz=timezone.utc), title="Evaluation creation date")
 
     def status(self, date: datetime = datetime.now(tz=timezone.utc)) -> Status:
         """DEPRECATED