update datastore export fields

GoogleCloudPlatform · Jan 4, 2024 · ecf10ba · ecf10ba
1 parent ab42c48
commit ecf10ba
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/retrieval_service/datastore/datastore.py b/retrieval_service/datastore/datastore.py
@@ -89,6 +89,20 @@ async def export_dataset(
                 "terminal",
                 "category",
                 "hour",
+                "sunday_start_hour",
+                "sunday_end_hour",
+                "monday_start_hour",
+                "monday_end_hour",
+                "tuesday_start_hour",
+                "tuesday_end_hour",
+                "wednesday_start_hour",
+                "wednesday_end_hour",
+                "thursday_start_hour",
+                "thursday_end_hour",
+                "friday_start_hour",
+                "friday_end_hour",
+                "saturday_start_hour",
+                "saturday_end_hour",
                 "content",
                 "embedding",
             ]