Merge pull request #566 from yuvipanda/injections

Tell pangeo-forge-runner what to inject
pangeo-forge · Oct 3, 2023 · 12f963b · 12f963b
2 parents 7e34030 + 07da347
commit 12f963b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/pangeo_forge_recipes/injections.py b/pangeo_forge_recipes/injections.py
@@ -0,0 +1,10 @@
+def get_injection_specs():
+    return {
+        "StoreToZarr": {
+            "target_root": "TARGET_STORAGE",
+        },
+        "WriteCombinedReference": {
+            "target_root": "TARGET_STORAGE",
+        },
+        "OpenURLWithFSSpec": {"cache": "INPUT_CACHE_STORAGE"},
+    }
diff --git a/pyproject.toml b/pyproject.toml
@@ -83,3 +83,6 @@ line_length = 100
 log_cli = false
 timeout = 30
 timeout_method = "signal"
+
+[project.entry-points."pangeo_forge_runner.injection_specs"]
+default_injections = "pangeo_forge_recipes.injections:get_injection_specs"