From 3811ea4258f930a17a3cfadce4f82ff17de5bae8 Mon Sep 17 00:00:00 2001
From: Joseph Xu <jzxu@google.com>
Date: Mon, 16 Dec 2024 00:11:26 -0800
Subject: [PATCH] Add CRS to inference GeoPackage output.

PiperOrigin-RevId: 706599023
---
 src/skai/model/inference_lib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/skai/model/inference_lib.py b/src/skai/model/inference_lib.py
index 4e6e32db..d6fb5838 100644
--- a/src/skai/model/inference_lib.py
+++ b/src/skai/model/inference_lib.py
@@ -632,7 +632,7 @@ def postprocess(
       else:
         geometries.append(shapely.wkt.loads(wkt))
     gdf = gpd.GeoDataFrame(
-        df.drop(columns=['footprint_wkt']), geometry=geometries
+        df.drop(columns=['footprint_wkt']), geometry=geometries, crs='EPSG:4326'
     )
     output_dir, output_file = os.path.split(output_path)
     gpkg_path = os.path.join(output_dir, f'{output_file}.gpkg')