add dry_run for load_ml_model Open-EO/openeo-geopyspark-driver#562

Open-EO · Apr 30, 2024 · a6a25a3 · a6a25a3
1 parent 7fa3a6b
commit a6a25a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/openeo_driver/ProcessGraphDeserializer.py b/openeo_driver/ProcessGraphDeserializer.py
@@ -777,6 +777,8 @@ def save_ml_model(args: dict, env: EvalEnv) -> MlModelResult:
 @process_registry_100.add_function(spec=read_spec("openeo-processes/experimental/load_ml_model.json"))
 @process_registry_2xx.add_function(spec=read_spec("openeo-processes/experimental/load_ml_model.json"))
 def load_ml_model(args: dict, env: EvalEnv) -> DriverMlModel:
+    if env.get(ENV_DRY_RUN_TRACER):
+        return DriverMlModel()
     job_id = extract_arg(args, "id")
     return env.backend_implementation.load_ml_model(job_id)