Merge pull request #562 from pangeo-forge/return-target-store

Return target_store from StoreToZarr stage
pangeo-forge · Aug 10, 2023 · 0ee97ff · 0ee97ff
2 parents 934e34b + 014df55
commit 0ee97ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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