Skip to content

Commit

Permalink
fix style flux-pmix
Browse files Browse the repository at this point in the history
Signed-off-by: vsoch <vsoch@users.noreply.github.com>
  • Loading branch information
vsoch committed Apr 7, 2023
1 parent 3721a69 commit 17514b4
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions packages/flux-pmix/package.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,16 +43,18 @@ def add_pluginpath(self):
pluginpath = join_path(self.prefix.lib, "flux/shell/plugins/pmix.so")
if spec.satisfies("@:0.3.0"):
rcfile = join_path(self.prefix.etc, "flux/shell/lua.d/mpi/openmpi@5.lua")
filter_file( r"pmix/pmix.so", pluginpath)
filter_file(r"pmix/pmix.so", pluginpath)
else:
rcdir = join_path(self.prefix.etc, "flux/shell/lua.d")
rcfile = join_path(rcdir, "pmix.lua")
mkdirp(rcdir)
with open(rcfile, "w") as fp:
fp.write("plugin.load(\"" + pluginpath + "\")")
fp.write('plugin.load("' + pluginpath + '")')

def setup_run_environment(self, env):
spec = self.spec
env.prepend_path("FLUX_SHELL_RC_PATH", join_path(self.prefix, "etc/flux/shell/lua.d"))
if spec.satisfies("@0.3.0:"):
env.prepend_path("FLUX_PMI_CLIENT_SEARCHPATH", join_path(self.prefix, "flux/upmi/plugins"))
env.prepend_path(
"FLUX_PMI_CLIENT_SEARCHPATH", join_path(self.prefix, "flux/upmi/plugins")
)

0 comments on commit 17514b4

Please sign in to comment.