fastapi is working now

allenai · Oct 17, 2024 · 02ab349 · 02ab349
1 parent 19421b5
commit 02ab349
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 49 deletions.
diff --git a/landsat/example_request.py b/landsat/example_request.py
@@ -7,7 +7,7 @@
 
 PORT = os.getenv("Landsat_PORT", default=5555)
 LANDSAT_ENDPOINT = f"http://localhost:{PORT}/detections"
-TIMEOUT_SECONDS = 60000
+TIMEOUT_SECONDS = 200000
 SCENE_ID = "LC09_L1GT_106084_20241002_20241002_02_T2"
 CROP_PATH = "/home/yawenz/rslearn_projects/landsat/temp_crops"
 SCRATCH_PATH = "/home/yawenz/rslearn_projects/landsat/temp_scratch"

diff --git a/rslp/api_main.py b/rslp/api_main.py
@@ -1,4 +1,4 @@
-"""VIIRS Vessel Detection Service."""
+"""Landsat Vessel Detection Service."""
 
 from __future__ import annotations
 
@@ -47,24 +47,10 @@ class LandsatRequest(BaseModel):
     scratch_path: str
     json_path: str
 
-    # class Config:
-    #     """example configuration for a request where files are stored in cloud"""
-
-    #     schema_extra = {
-    #         "example": {
-    #             "input_dir": "input",
-    #             "output_dir": "output",
-    #             "filename": "VJ102DNB.A2022362.0154.021.2022362055600.nc",
-    #             "geo_filename": "VJ103DNB.A2022362.0154.021.2022362052511.nc",
-    #             "modraw_filename": "VJ102MOD.A2022362.0154.002.2022362115107.nc",
-    #             "modgeo_filename": "VJ103MOD.A2022362.0154.002.2022362095104.nc",
-    #         },
-    # }
-
 
 @app.on_event("startup")
 async def rslp_init() -> None:
-    """VIIRS Vessel Service Initialization."""
+    """Landsat Vessel Service Initialization."""
     logger.info("Initializing")
 
 
@@ -90,38 +76,6 @@ async def get_detections(info: LandsatRequest, response: Response) -> LandsatRes
         logger.error(f"Error during prediction pipeline: {e}")
         return LandsatResponse(status=["error"], predictions=[])
 
-    #     ves_detections = all_detections["vessel_detections"]
-
-    #     satellite_name = utils.get_provider_name(dnb_dataset)
-    #     acquisition_time, end_time = utils.get_acquisition_time(dnb_dataset)
-    #     chips_dict = utils.get_chips(image, ves_detections, dnb_dataset)
-    #     if info.gcp_bucket is not None:
-    #         chips_dict = utils.upload_image(
-    #             info.gcp_bucket, chips_dict, info.output_dir, dnb_path
-    #         )
-    #     else:
-    #         utils.save_chips_locally(
-    #             chips_dict,
-    #             destination_path=info.output_dir,
-    #             chip_features=ves_detections,
-    #         )
-
-    #     average_moonlight = RoundedFloat(utils.get_average_moonlight(dnb_dataset), 2)
-
-    #     frame_extents = utils.get_frame_extents(dnb_dataset)
-
-    # predictions = utils.format_detections(chips_dict)
-    # time_s = round(perf_counter() - start)
-    # n_ves = len(chips_dict)
-    # logger.info(
-    #     f"In frame: {dnb_path}, vvd detected {n_ves} vessels in ({time_s} seconds)"
-    # )
-    # response.headers["n_detections"] = str(len(chips_dict))
-    # response.headers["avg_moonlight"] = str(average_moonlight)
-    # response.headers["lightning_count"] = str(all_detections["lightning_count"])
-    # response.headers["gas_flare_count"] = str(all_detections["gas_flare_count"])
-    # response.headers["inference_time"] = str("elapsed_time")
-
 
 if __name__ == "__main__":
     uvicorn.run("api_main:app", host=HOST, port=PORT, proxy_headers=True)