change drop_vars in run.py

ClimateImpactLab · Jan 10, 2025 · 456c50e · 456c50e
1 parent b9b6349
commit 456c50e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyCIAM/run.py b/pyCIAM/run.py
@@ -1425,7 +1425,6 @@ def execute_pyciam(
     # Rechunk and save final
     ###############################
     client.gather(ciam_futs_2.tolist())
-    assert [f.status == "finished" for f in ciam_futs_2.tolist()]
     client.cancel(ciam_futs_2)
     del ciam_futs_2
 
@@ -1437,7 +1436,7 @@ def execute_pyciam(
             storage_options=storage_options,
             chunks={"case": -1, seg_var: this_chunksize},
         )
-        .drop("npv")
+        .drop_vars("npv")
         .chunk({"year": 10})
         .persist()
     )