fixup

SciML · Oct 3, 2023 · eddd2c8 · eddd2c8
1 parent d3e2b46
commit eddd2c8
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/test/python/pycall.jl b/test/python/pycall.jl
@@ -6,12 +6,18 @@ function with_pycall_in_default_environment(f)
     path = Pkg.project().path
     Pkg.activate()
     install = "PyCall" ∉ keys(Pkg.project().dependencies)
-    install && Pkg.add("PyCall")
+    if install
+        Pkg.add("PyCall")
+    end
+    Pkg.activate(path)
     try
         f()
     finally
-        install && Pkg.rm("PyCall")
-        Pkg.activate(path)
+        if install
+            Pkg.activate()
+            Pkg.rm("PyCall")
+            Pkg.activate(path)
+        end
     end
 end