refactor: extract function

DanielSchiavini · Oct 9, 2024 · f2e1da7 · f2e1da7
1 parent b98fe9d
commit f2e1da7
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/boa/__init__.py b/boa/__init__.py
@@ -41,16 +41,16 @@ def swap_env(new_env):
         yield
 
 
-def set_env(new_env):
+def _set_env(new):
     global env
-    get_env = lambda: env  # noqa: E731
+    env = new
+    Env._singleton = new
 
-    def setter(new):
-        global env
-        env = new
-        Env._singleton = new
 
-    return Open(get_env, setter, new_env)
+def set_env(new_env):
+    global env
+    get_env = lambda: env  # noqa: E731
+    return Open(get_env, _set_env, new_env)
 
 
 def fork(