update calling write_flowveldepth functions

shorvath-noaa · Feb 29, 2024 · 4a43974 · 4a43974
1 parent 693b9a2
commit 4a43974
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/src/troute-nwm/src/nwm_routing/output.py b/src/troute-nwm/src/nwm_routing/output.py
@@ -202,15 +202,18 @@ def nwm_output_generator(
 
         nudge = np.concatenate([r[8] for r in results])
         usgs_positions_id = np.concatenate([r[3][0] for r in results])
-        nhd_io.write_flowveldepth_netcdf(Path(stream_output_directory), 
-                                         flowveldepth, 
-                                         nudge, 
-                                         usgs_positions_id, 
-                                         t0, 
-                                         int(stream_output_timediff), 
-                                         stream_output_type,
-                                         stream_output_internal_frequency,
-                                         cpu_pool = cpu_pool)
+        nhd_io.write_flowveldepth(
+            Path(stream_output_directory), 
+            flowveldepth, 
+            nudge, 
+            usgs_positions_id, 
+            t0, 
+            dt,
+            int(stream_output_timediff), 
+            stream_output_type,
+            stream_output_internal_frequency,
+            cpu_pool = cpu_pool
+            )
 
     if test:
         flowveldepth.to_pickle(Path(test))