From e85be3787d4f5b262939baa2ee93e507c26fc264 Mon Sep 17 00:00:00 2001
From: devsjc <sol@openclimatefix.org>
Date: Fri, 6 Oct 2023 15:09:34 +0100
Subject: [PATCH] Update variable handling (stop using reserved keyword)

---
 nwp/assets/ecmwf/mars.py | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/nwp/assets/ecmwf/mars.py b/nwp/assets/ecmwf/mars.py
index 1992e0e..1daa1f3 100644
--- a/nwp/assets/ecmwf/mars.py
+++ b/nwp/assets/ecmwf/mars.py
@@ -6,20 +6,19 @@
 
 
 @contextlib.contextmanager
-def modify_env(vars: dict[str, str]):
+def modify_env(newvars: dict[str, str]):
     """Temporarily modify the environment."""
-    oldvars = os.environ.copy()
-    for var in vars:
-        newval = vars[var]
-        os.environ[var] = newval
+    oldvars: dict[str, str] = os.environ.copy()
+    for key in newvars:
+        os.environ[key] = newvars[key]
     try:
         yield
     finally:
-        for var in vars:
-            if var in oldvars:
-                os.environ[var] = oldvars[var]
+        for key in newvars:
+            if key in oldvars:
+                os.environ[key] = oldvars[key]
             else:
-                del os.environ[var]
+                del os.environ[key]
 
 class NWPConsumerConfig(Config):
     """Configuration for the NWP consumer."""