fix: use custom field not str

CAVEconnectome · Aug 29, 2024 · 1c91224 · 1c91224
1 parent 19ebf4b
commit 1c91224
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/materializationengine/schemas.py b/materializationengine/schemas.py
@@ -70,24 +70,21 @@ class TaskParamsSchema(Schema):
 
 class CeleryBeatSchema(Schema):
     name = fields.Str(required=True, metadata={"description": "Name of the task"})
-    minute = fields.Str(
+    minute = CronField(
         default="*", metadata={"description": "Minute field for cron schedule"}
     )
-    hour = fields.Str(
+    hour = CronField(
         default="*", metadata={"description": "Hour field for cron schedule"}
     )
-    day_of_week = fields.Str(
-        required=False,
+    day_of_week = CronField(
         default="*",
         metadata={"description": "Day of week for cron schedule"},
     )
-    day_of_month = fields.Str(
-        required=False,
+    day_of_month = CronField(
         default="*",
         metadata={"description": "Day of month for cron schedule"},
     )
-    month_of_year = fields.Str(
-        required=False,
+    month_of_year = CronField(
         default="*",
         metadata={"description": "Month of year for cron schedule"},
     )