Add None as ArgType

ekiefl · Dec 31, 2024 · 407509b · 407509b
1 parent c48dd43
commit 407509b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pooltool/physics/resolve/types.py b/pooltool/physics/resolve/types.py
@@ -3,7 +3,7 @@
 from pooltool.serialize import conversion
 from pooltool.serialize.serializers import SerializeFormat
 
-ArgType = Union[float, int, str, bool]
+ArgType = Union[float, int, str, bool, None]
 """Allowable types model arguments"""
 
 # Recursive type definition is not fixed with PEP 563, so ForwardRef type is required.

diff --git a/tests/physics/resolve/test_types.py b/tests/physics/resolve/test_types.py
@@ -9,6 +9,7 @@ def test_model_args_flat(tmp_path):
         "a": 42,
         "b": True,
         "c": "string",
+        "d": None,
     }
     conversion.unstructure_to(d, tmp_path / "test.yaml")
     assert d == conversion.structure_from(tmp_path / "test.yaml", ModelArgs)