Added utm in setup and close() for dataset in hycom

schism-dev · Jun 10, 2022 · bac8635 · bac8635
1 parent 01b03f1
commit bac8635
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pyschism/forcing/hycom/hycom2schism.py b/pyschism/forcing/hycom/hycom2schism.py
@@ -414,6 +414,8 @@ def fetch_data(self, outdir: Union[str, os.PathLike], start_date, rnday, elev2D=
                     dst_uv['time_series'][it,ind1:ind2,:,1] = vvel_int
                     #timeseries_uv[it,:,:,1]=vvel_int
 
+                ds.close()
+
         logger.info(f'Writing *th.nc takes {time()-t0} seconds')
 
 class Nudge:
@@ -627,6 +629,8 @@ def fetch_data(self, outdir: Union[str, os.PathLike], hgrid, vgrid, start_date,
             temp_int = interp_to_points_3d(dep, y2, x2, bxyz, ptemp)
             temp_int = temp_int.reshape(zcor2.shape)
             timeseries_t[it,:,:,0]=temp_int
+
+            ds.close()
 
         with Dataset(outdir / 'TEM_nu.nc', 'w', format='NETCDF4') as dst:
         #dimensions

diff --git a/setup.py b/setup.py
@@ -122,7 +122,8 @@ def run(self):
         'cfgrib',
         'zarr',
         'fsspec',
-        'stormevents'
+        'stormevents',
+        'utm',
         # 'dask_geopandas @ git+git://github.com/geopandas/dask-geopandas.git@master',
     ],
     entry_points={'console_scripts': ['pyschism = pyschism.__main__:main']},