From 456c50e0e0e4a47969a033407eb60f31812aabd8 Mon Sep 17 00:00:00 2001
From: Ian Bolliger <ian@reask.earth>
Date: Fri, 10 Jan 2025 00:27:47 +0000
Subject: [PATCH] change drop_vars in run.py

---
 pyCIAM/run.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pyCIAM/run.py b/pyCIAM/run.py
index 873f911..a353f75 100644
--- 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()
     )