Correcting a mistake--need to return an OrchestratorResult object, or…

… else the task is failed
SkyTruth · Nov 9, 2023 · 789240b · 789240b
1 parent 5ba02d7
commit 789240b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/cerulean_cloud/cloud_run_orchestrator/handler.py b/cerulean_cloud/cloud_run_orchestrator/handler.py
@@ -548,4 +548,4 @@ async def _orchestrate(
             del out_fc_offset_2
         else:
             print(f"{start_time}: WARNING: Operating as a DRY RUN!!")
-    return "Success"
+    return OrchestratorResult(status="Success")
diff --git a/cerulean_cloud/cloud_run_orchestrator/schema.py b/cerulean_cloud/cloud_run_orchestrator/schema.py
@@ -1,7 +1,7 @@
 """schema for orchestration enpoint"""
 from typing import Optional
 
-import geojson
+# import geojson
 from pydantic import BaseModel
 
 
@@ -22,8 +22,9 @@ class OrchestratorResult(BaseModel):
     orchestrator result from the model
     """
 
-    classification_base: geojson.FeatureCollection
-    classification_offset: geojson.FeatureCollection
-    classification_merged: geojson.FeatureCollection
-    ntiles: int
-    noffsettiles: int
+    # classification_base: geojson.FeatureCollection
+    # classification_offset: geojson.FeatureCollection
+    # classification_merged: geojson.FeatureCollection
+    # ntiles: int
+    # noffsettiles: int
+    status: str