From 014df559b05c18cb23ccb337055cd22d60af60c9 Mon Sep 17 00:00:00 2001 From: Julius Busecke Date: Thu, 10 Aug 2023 17:44:34 +0200 Subject: [PATCH] Return target_store from StoreToZarr stage Superseeds https://github.com/jbusecke/pangeo-forge-recipes/pull/1 (see over there for discussion). --- pangeo_forge_recipes/transforms.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pangeo_forge_recipes/transforms.py b/pangeo_forge_recipes/transforms.py index bd23db66..0d050923 100644 --- a/pangeo_forge_recipes/transforms.py +++ b/pangeo_forge_recipes/transforms.py @@ -399,4 +399,5 @@ def expand(self, datasets: beam.PCollection) -> beam.PCollection: target_store = schema | PrepareZarrTarget( target=self.get_full_target(), target_chunks=self.target_chunks ) - return rechunked_datasets | StoreDatasetFragments(target_store=target_store) + rechunked_datasets | StoreDatasetFragments(target_store=target_store) + return target_store