Принципиальное условное наследование в Python ⇐ Python

[Anonymous]

У меня есть класс, который наследует np.ndarray:

Код: Выделить всё

class A(np.ndarray):
def __new__(cls, foo: np.ndarray):
# Operate on foo...
x = np.array([1.]) + foo # (placeholder)

return x

def bar(self, x):
# Other stuff...

Я хочу добавить поведение, независимое от процессора/графического процессора, определяя, является ли вход np.ndarray или cupy.ndarray, и динамически наследуя один из два, вот так:

Код: Выделить всё

class A:
def __new__(cls, foo: Union[np.ndarray, cupy.ndarray]):
xp = cupy.get_array_module(foo) # Returns either `numpy` or `cupy` depending on `foo`

# Operate on foo...
x = xp.array([1.]) + foo # (placeholder)

return xp.__new__(cls, x)

def bar(self, x):
# Other stuff...

Я думаю, что следующее может сработать, но это очень запутанно:

Код: Выделить всё

class BaseA(ABC):
def bar(self, x):
# Other stuff...

class ANumpy(BaseA, np.ndarray):
pass

class ACupy(BaseA, cupy.ndarray):
pass

class A:
def __new__(cls, foo: Union[np.ndarray, cupy.ndarray]):
if isinstance(foo, cupy.ndarray):
return ACupy(foo)

return ANumpy(foo)

Как мне это сделать правильно/хорошо?


Подробнее здесь: https://stackoverflow.com/questions/792 ... -in-python