Skip to content

Commit

Permalink
minor changes to pipeline #3
Browse files Browse the repository at this point in the history
  • Loading branch information
mattjbr123 committed Sep 25, 2024
1 parent e4ec977 commit 438169e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/convert_GEAR_beam.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
td = "/work/scratch-pw2/mattjbr"
tn = "gear_1hrly_fulloutput_yearly_100km_chunks.zarr"
target_chunks = {"time": int(365.25*24), "y": 100, "x": 100, "bnds": 2}
nprocs = 8
nprocs = 4
prune = 12 # no. of files to process, set to 0 to use all

if not os.path.exists(td):
Expand Down Expand Up @@ -86,7 +86,7 @@ def expand(self, pcoll: beam.PCollection) -> beam.PCollection:
recipe = (
beam.Create(pattern.items())
| OpenWithXarray(file_type=pattern.file_type)
| DataVarToCoordVar() # the preprocess
| DataVarToCoordVar()
| StoreToZarr(
target_root=td,
store_name=tn,
Expand Down

0 comments on commit 438169e

Please sign in to comment.