Guard against all name (#1437)

Quantum-Accelerators · Dec 29, 2023 · 9354357 · 9354357
1 parent 9a93a6d
commit 9354357
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/quacc/wflow_tools/customizers.py b/src/quacc/wflow_tools/customizers.py
@@ -122,6 +122,8 @@ def customize_funcs(
 
     funcs_dict = dict(zip(names, funcs))
 
+    if "all" in names:
+        raise ValueError("Invalid function name: 'all' is a reserved name.")
     if bad_decorator_keys := [k for k in decorators if k not in names and k != "all"]:
         raise ValueError(
             f"Invalid decorator keys: {bad_decorator_keys}. " f"Valid keys are: {names}"

diff --git a/tests/core/wflow/test_customizers.py b/tests/core/wflow/test_customizers.py
@@ -30,3 +30,6 @@ def mult(a, b=1, c=2, d=2):
 
     with pytest.raises(ValueError):
         customize_funcs(["add", "mult"], [add, mult], parameters={"bad": {"b": 2}})
+
+    with pytest.raises(ValueError):
+        customize_funcs("all", [add], parameters={"all": {"b": 2}})