From d7e8f61e76af7ea47885576039b2e22be382cbf4 Mon Sep 17 00:00:00 2001 From: Harpo Harbert Date: Fri, 8 Sep 2023 10:39:44 -0700 Subject: [PATCH] Fixes date serializer --- chowda/fields.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/chowda/fields.py b/chowda/fields.py index 4d005c5a..20dc582d 100644 --- a/chowda/fields.py +++ b/chowda/fields.py @@ -1,4 +1,5 @@ from dataclasses import dataclass +from datetime import datetime from typing import Any from starlette.datastructures import FormData @@ -101,6 +102,19 @@ async def parse_obj(self, request: Request, obj: Any) -> Any: return [run.dict() for run in new_runs or obj.metaflow_runs] + async def serialize_value( + self, request: Request, value: Any, action: RequestAction + ) -> Any: + return [ + { + **run, + 'finished_at': run['finished_at'].isoformat() + if run.get('finished_at') + else None, + } + for run in value + ] + @dataclass class BatchPercentCompleted(BaseField):