From 661f756fff4da279573c70e45e9c361ff6cc34ab Mon Sep 17 00:00:00 2001 From: Jan Janssen Date: Fri, 15 Nov 2024 22:08:18 +0100 Subject: [PATCH] fix input checks --- executorlib/standalone/inputcheck.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/executorlib/standalone/inputcheck.py b/executorlib/standalone/inputcheck.py index a78e2e8d..5abea87b 100644 --- a/executorlib/standalone/inputcheck.py +++ b/executorlib/standalone/inputcheck.py @@ -117,9 +117,9 @@ def check_pmi(backend: str, pmi: Optional[str]) -> None: """ Check if pmi is valid for the selected backend and raise a ValueError if it is not. """ - if backend != "flux" and pmi is not None: + if backend != "flux_allocation" and pmi is not None: raise ValueError("The pmi parameter is currently only implemented for flux.") - elif backend == "flux" and pmi not in ["pmix", "pmi1", "pmi2", None]: + elif backend == "flux_allocation" and pmi not in ["pmix", "pmi1", "pmi2", None]: raise ValueError( "The pmi parameter supports [pmix, pmi1, pmi2], but not: " + pmi )