From 8e1405af80b9d9041ba1da7806e6c694bf6a8f85 Mon Sep 17 00:00:00 2001 From: Frederik Hennig <frederik.hennig@fau.de> Date: Wed, 29 Jan 2025 14:17:22 +0100 Subject: [PATCH] fix mutable defaults error with Category descriptor and Python 3.12 --- noxfile.py | 2 +- src/pystencils/codegen/config.py | 12 ++++++++---- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/noxfile.py b/noxfile.py index 1b882478a..11b3731ec 100644 --- a/noxfile.py +++ b/noxfile.py @@ -86,8 +86,8 @@ def typecheck(session: nox.Session): session.run("mypy", "src/pystencils") +@nox.session(python=["3.10", "3.12", "3.13"], tags=["test"]) @nox.parametrize("cupy_version", [None, "12", "13"], ids=["cpu", "cupy12", "cupy13"]) -@nox.session(python="3.10", tags=["test"]) def testsuite(session: nox.Session, cupy_version: str | None): if cupy_version is not None: install_cupy(session, cupy_version, skip_if_no_cuda=True) diff --git a/src/pystencils/codegen/config.py b/src/pystencils/codegen/config.py index cbb3f4f32..d56fe6642 100644 --- a/src/pystencils/codegen/config.py +++ b/src/pystencils/codegen/config.py @@ -196,12 +196,16 @@ class Category(Generic[Category_T]): def __get__(self, obj: ConfigBase, objtype: type[ConfigBase] | None = None) -> Category_T: if obj is None: - return self._default + return None - return cast(Category_T, getattr(obj, self._lookup, None)) + cat = getattr(obj, self._lookup, None) + if cat is None: + cat = self._default.copy() + setattr(obj, self._lookup, cat) + return cast(Category_T, cat) - def __set__(self, obj: ConfigBase, cat: Category_T): - setattr(obj, self._lookup, cat.copy()) + def __set__(self, obj: ConfigBase, cat: Category_T | None): + setattr(obj, self._lookup, cat.copy() if cat is not None else None) class _AUTO_TYPE: ... # noqa: E701 -- GitLab