log final stream number

geoglows · Dec 21, 2023 · c0443bf · c0443bf
1 parent 3605f14
commit c0443bf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/run_5_global_geometry.py b/run_5_global_geometry.py
@@ -0,0 +1,10 @@
+import geopandas as gpd
+import pandas as pd
+import glob
+import os
+import natsort
+
+
+gpq = natsort.natsorted(glob.glob('/Volumes/T9Hales4TB/geoglows2/tdxhydro-inputs/*/*geoparquet'))
+gdf = pd.concat([gpd.read_parquet(g).simplify() for g in gpq])
+print(gpq)
diff --git a/tdxhydrorapid/inputs.py b/tdxhydrorapid/inputs.py
@@ -218,6 +218,8 @@ def rapid_master_files(streams_gpq: str,
         sgdf.loc[sgdf[id_field].isin(list(nx.ancestors(G, term_node)) + [term_node, ]), 'TerminalLink'] = term_node
     sgdf['TerminalLink'] = sgdf['TerminalLink'].astype(int)
 
+    logger.info(f'Final Stream Count: {sgdf.shape[0]}')
+
     if cache_geometry:
         logger.info('\tWriting altered geometry to geopackage')
         region_number = sgdf['TDXHydroRegion'].values[0]