From 9f87a2509dbc31cd5a3be9711c21b3257261d946 Mon Sep 17 00:00:00 2001 From: Alexandru Fikl Date: Fri, 25 Jul 2025 20:30:50 +0300 Subject: [PATCH] feat: remove dataclass from BaseFakeNumpyNamespace --- arraycontext/fake_numpy.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py index b7b47017..b4567d11 100644 --- a/arraycontext/fake_numpy.py +++ b/arraycontext/fake_numpy.py @@ -29,7 +29,6 @@ import operator from abc import ABC, abstractmethod -from dataclasses import dataclass from typing import TYPE_CHECKING, Any, Literal, cast, overload import numpy as np @@ -66,14 +65,13 @@ # {{{ BaseFakeNumpyNamespace -@dataclass(frozen=True) class BaseFakeNumpyNamespace(ABC): _array_context: ArrayContext linalg: BaseFakeNumpyLinalgNamespace def __init__(self, array_context: ArrayContext): - object.__setattr__(self, "_array_context", array_context) - object.__setattr__(self, "linalg", self._get_fake_numpy_linalg_namespace()) + self._array_context = array_context + self.linalg = self._get_fake_numpy_linalg_namespace() def _get_fake_numpy_linalg_namespace(self): return BaseFakeNumpyLinalgNamespace(self._array_context)