fix UT

ray-project · Feb 25, 2022 · 37698a6 · 37698a6
1 parent 1b409f1
commit 37698a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ray/tests/test_runtime_env_plugin.py b/python/ray/tests/test_runtime_env_plugin.py
@@ -105,9 +105,10 @@ def modify_context(
 
 
 def test_plugin_hang(ray_start_regular):
+    env_key = MyPluginForHang.env_key
     @ray.remote(num_cpus=0.1)
     def f():
-        return os.environ[MyPluginForHang.env_key]
+        return os.environ[env_key]
 
     refs = [
         f.options(