Extensions: fixed the left join (#19)

* changed join as the left/right were incorret * linting
taddyb · Apr 3, 2024 · 9b930cd · 9b930cd
1 parent aa2a581
commit 9b930cd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/marquette/conf/config.yaml b/marquette/conf/config.yaml
@@ -1,6 +1,6 @@
 name: MERIT
 data_path: /projects/mhpi/data/${name}
-zone: 72
+zone: 75
 create_edges:
   buffer: 0.3334
   dx: 2000

diff --git a/marquette/merit/extensions.py b/marquette/merit/extensions.py
@@ -116,8 +116,11 @@ def soils_data(cfg: DictConfig, edges: zarr.Group) -> None:
         )
         .otherwise(pl.col(["OM_05_M_25", "Cl_05_Mn", "Sd_05_Mn", "St_05_Mn"]))
     )
+    graph_cols = ["COMID", "up1", "NextDownID"]
+    df_cols = graph_cols + attributes
+    _df = df_filled.select(pl.col(df_cols))
     edges_df = pl.DataFrame({"COMID": edges.merit_basin[:]})
-    joined_df = df_filled.join(edges_df, on="COMID", how="left", join_nulls=True)
+    joined_df = edges_df.join(_df, on="COMID", how="left", join_nulls=True)
     for i in range(len(names)):
         edges.array(
             name=names[i],