pythran dataclass

my_pythran_file.py

from dataclasses import dataclass


@dataclass
class Base:
    a: "float[:, :]"
    b: "float[:, :]"
    c: "float[:, :]"

    def intermediate(self, d: int):
        print("intermediate Base class")
        return self.a ** d

    def compute(self, e: int):
        return self.intermediate(e) + self.b * self.c

now_in_python.py

import numpy as np


class Child(Base):
    def __init__(self, myinput):

        a = b = c = myinput * np.ones((2, 2))

        super().__init__(a, b, c)

    def intermediate(self, d):
        print("intermediate Child class")
        return self.b ** d


if __name__ == "__main__":
    child = Child(1)
    child.compute(1)