From 7347bebe3c8a9c8b895a2e1061bfddac0b96135f Mon Sep 17 00:00:00 2001
From: ValentinaHutter <85164505+ValentinaHutter@users.noreply.github.com>
Date: Tue, 4 Feb 2025 15:19:23 +0100
Subject: [PATCH] add pystac to dependencies (#317)

* add pystac to dependencies

* update dask geopandas
---
 .../process_implementations/ml/random_forest.py                | 2 +-
 pyproject.toml                                                 | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/openeo_processes_dask/process_implementations/ml/random_forest.py b/openeo_processes_dask/process_implementations/ml/random_forest.py
index 498b4d7a..de248b64 100644
--- a/openeo_processes_dask/process_implementations/ml/random_forest.py
+++ b/openeo_processes_dask/process_implementations/ml/random_forest.py
@@ -95,7 +95,7 @@ def drop_col(df, keep_var):
     if isinstance(predictors, gpd.GeoDataFrame):
         predictors = dask_geopandas.from_geopandas(predictors, npartitions=1)
 
-    if isinstance(predictors, dask_geopandas.core.GeoDataFrame):
+    if isinstance(predictors, dask_geopandas.expr.GeoDataFrame):
         data_ddf = (
             predictors.to_dask_dataframe().reset_index().repartition(npartitions=1)
         )
diff --git a/pyproject.toml b/pyproject.toml
index 1775e5fb..f90ffc2f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ pandas = { version = ">=2.0.0", optional = true }
 xarray = { version = ">=2022.11.0,<=2024.3.0", optional = true }
 dask = {extras = ["array", "dataframe", "distributed"], version = ">=2023.4.0", optional = true}
 rasterio = { version = "^1.3.4", optional = true }
-dask-geopandas = { version = ">=0.2.0,<1", optional = true }
+dask-geopandas = { version = "0.4.3", optional = true }
 xgboost = { version = ">=1.5.1", optional = true }
 rioxarray = { version = ">=0.12.0,<1", optional = true }
 openeo-pg-parser-networkx = { version = ">=2024.7", optional = true }
@@ -44,6 +44,7 @@ joblib = { version = ">=1.3.2", optional = true }
 geoparquet = "^0.0.3"
 pyarrow = "^15.0.2"
 numpy = { version = "<2.0.0", optional = false }
+pystac = { version = "<1.12.0", optional = false }
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.2.0"