Merge pull request #81 from WorldCereal/batch_size

kvantricht · web-flow · commit b479fb339ffe · 2024-06-25T12:06:49.000+02:00
batch_size as parameter
diff --git a/src/worldcereal/openeo/feature_extractor.py b/src/worldcereal/openeo/feature_extractor.py
@@ -110,9 +110,11 @@ def execute(self, inarr: xr.DataArray) -> xr.DataArray:
             get_presto_features,
         )
 
+        batch_size = self._parameters.get("batch_size", 4096)
+
         self.logger.info("Extracting presto features")
         features = get_presto_features(
-            inarr, presto_model_url, self.epsg, batch_size=4096
+            inarr, presto_model_url, self.epsg, batch_size=batch_size
         )
         return features
 

Original file line number	Diff line number	Diff line change
`@@ -110,9 +110,11 @@ def execute(self, inarr: xr.DataArray) -> xr.DataArray:`
`110`	`110`	`get_presto_features,`
`111`	`111`	`)`
`112`	`112`
	`113`	`+ batch_size = self._parameters.get("batch_size", 4096)`
	`114`	`+`
`113`	`115`	`self.logger.info("Extracting presto features")`
`114`	`116`	`features = get_presto_features(`
`115`		`- inarr, presto_model_url, self.epsg, batch_size=4096`
	`117`	`+ inarr, presto_model_url, self.epsg, batch_size=batch_size`
`116`	`118`	`)`
`117`	`119`	`return features`
`118`	`120`