fix: include index columns for geodataframes

pipeline/resources/postgis_geopandas_io_manager.py

@@ -171,8 +171,8 @@ def handle_output(self, context: OutputContext, obj: geopandas.GeoDataFrame):
         schema, table = self._get_schema_table(context.asset_key)
 
         if isinstance(obj, geopandas.GeoDataFrame):
-            len(obj)
             with connect_postgresql(config=self._config) as con:
+                self._create_schema_if_not_exists(schema, con)
                 if context.has_partition_key:
                     # add additional column (name? for now just partition)
                     # to the frame and initialize with partition_name
@@ -189,9 +189,9 @@ def handle_output(self, context: OutputContext, obj: geopandas.GeoDataFrame):
                     # All data can be replaced (e.g. deleted before insertion).
                     # geopandas will take care of this.
                     if_exists_action = 'replace'
-
-                self._create_schema_if_not_exists(schema, con)
-                obj.to_postgis(con=con, name=table, schema=schema, if_exists=if_exists_action, chunksize=self.chunksize)
+                obj.to_postgis(
+                    con=con, name=table, index=True, schema=schema, if_exists=if_exists_action, chunksize=self.chunksize
+                )
                 context.add_output_metadata({'num_rows': len(obj), 'table_name': f'{schema}.{table}'})
         else:
             super().handle_output(context, obj)