Cannot assign an istance of custom class into a numpy array

[Bleed00]

I created a simple program that generates mazes and uses pygame to display the output.
It looks cool and all but when scaling the maze (making it, for example, 64x64) takes lots of time to generate and I wanted to use Codon to speedup this project.
The code works perfectly in standard python, I tried following the documentation but I can't get over this error:

codon run to_comp.py 
to_comp.py:83:29-47: error: 'cell' object has no attribute '__to_py__'
╰─ to_comp.py:153:11-39: error: during the realization of map_generation(WIDTH: int, HEIGTH: int)

The code is the following:

import python
from python import numpy as np
from collections import deque as dq
import time
import random as rand
from python import pygame
#colors
BLACK = (0,0,0)
WHITE = (255,255,255)
RED = (255,0,0)
GREEN = (0,255,0)
#dimensioni
WINDOW_WIDTH = 608
WINDOW_HEIGHT = 608
clock = pygame.time.Clock()
screen = pygame.display.set_mode((WINDOW_WIDTH, WINDOW_HEIGHT))
pygame.mouse.set_visible(1)
pygame.display.set_caption('Labirinto')
#SINGLE CELL CLASS
class cell:
    visited: int
    UP : int
    DOWN : int
    RIGHT : int
    LEFT : int
    direction : int
    def __init__(self):
        self.visited = 0 
        self.UP = 1
        self.DOWN = 1
        self.RIGHT = 1
        self.LEFT = 1
    def used(self):
        self.visited = 1
    def change_dir(self, direction):
        if direction == 2:
            self.DOWN = 0
        if direction == 3:
            self.UP = 0
        if direction == 0:
            self.RIGHT = 0
        if direction == 1:
            self.LEFT = 0
def drawOutside(width,heigth):
        pygame.draw.line(screen,WHITE,(0,0),(0,heigth),3)
        pygame.draw.line(screen,WHITE,(0,0),(width,0),3)
        pygame.draw.line(screen,WHITE,(width,heigth-1),(0,heigth-1),3)
        pygame.draw.line(screen,WHITE,(width-1,heigth),(width-1,0),3)

def drawGrid(n_X,n_Y,map):
        i = 0
        for x in range(0, WINDOW_WIDTH, int(WINDOW_WIDTH/n_X)):
            j = 0
            for y in range(0, WINDOW_HEIGHT, int(WINDOW_HEIGHT/n_Y)):
                if map[i][j].visited == 1:
                    if map[i][j].UP == 1:
                        pygame.draw.line(screen, WHITE, [x,y],[x + int(WINDOW_WIDTH/n_X),y], 1)
                    if map[i][j].DOWN == 1:
                        pygame.draw.line(screen, WHITE, [x,y + int(WINDOW_HEIGHT/n_Y)],[x + int(WINDOW_WIDTH/n_X),y + int(WINDOW_HEIGHT/n_Y)], 1)
                    if map[i][j].RIGHT == 1:
                        pygame.draw.line(screen, WHITE, [x + int(WINDOW_WIDTH/n_X),y],[x + int(WINDOW_WIDTH/n_X) ,y + int(WINDOW_HEIGHT/n_Y)], 1)
                    if map[i][j].LEFT == 1:
                        pygame.draw.line(screen, WHITE, [x,y],[x,y + int(WINDOW_HEIGHT/n_Y)], 1)
                j+=1
            pygame.display.flip()
            i+=1
def map_generation(WIDTH, HEIGTH):
    map = np.ndarray((WIDTH,HEIGTH), dtype=cell)

    for i in range(WIDTH):
            for j in range(HEIGTH):
                map[i][j] = cell().__to_py__()
    return map

def map_routing(map,HEIGTH,WIDTH):
    stack = dq()
    stack.append((0,0))
    current = dq()
    while len(stack) != 0:
        clock.tick(60)
        screen.fill(BLACK)
        drawOutside(WINDOW_WIDTH,WINDOW_HEIGHT)
        current.append(stack[-1])
        arr = np.array([0,0,0,0],dtype=bool)
        if current[-1][0] + 1 < HEIGTH and map[current[-1][0] + 1][current[-1][1]].visited == 0 : arr[0] = True
        if map[current[-1][0] - 1][current[-1][1]].visited == 0 and current[-1][0] - 1 != -1 : arr[1] = True
        if current[-1][1] + 1 < WIDTH and map[current[-1][0]][current[-1][1] + 1].visited == 0 : arr[2] = True
        if map[current[-1][0]][current[-1][1] - 1].visited == 0 and current[-1][1] - 1 != -1 : arr[3] = True
        exists = 1 in arr
        if exists:
            choice = rand.randrange(0,4)
            while exists == 1:
                if arr[choice] == 1:
                    exists = False
                    match choice:
                        case 0:
                            map[current[-1][0]][current[-1][1]].used()
                            map[current[-1][0] + 1][current[-1][1]].used()
                            map[current[-1][0]][current[-1][1]].change_dir(0)
                            map[current[-1][0] + 1][current[-1][1]].change_dir(1)
                            stack.append((current[-1][0] + 1,current[-1][1]))
                        case 1:
                            map[current[-1][0]][current[-1][1]].used()
                            map[current[-1][0] - 1][current[-1][1]].used()
                            map[current[-1][0]][current[-1][1]].change_dir(1)
                            map[current[-1][0] - 1][current[-1][1]].change_dir(0)
                            stack.append((current[-1][0] - 1,current[-1][1]))
                        case 2:
                            map[current[-1][0]][current[-1][1]].used()
                            map[current[-1][0]][current[-1][1] + 1].used()
                            map[current[-1][0]][current[-1][1]].change_dir(2)
                            map[current[-1][0]][current[-1][1] + 1].change_dir(3)
                            stack.append((current[-1][0],current[-1][1] + 1))
                        case 3:
                            map[current[-1][0]][current[-1][1]].used()
                            map[current[-1][0]][current[-1][1] - 1].used()
                            map[current[-1][0]][current[-1][1]].change_dir(3)
                            map[current[-1][0]][current[-1][1] - 1].change_dir(2)
                            stack.append((current[-1][0],current[-1][1] - 1))
                else:
                    choice = rand.randrange(0,4)
            
            drawGrid(WIDTH,HEIGTH,map)
        else:
            stack.pop()

    return map




pygame.init()



while 1:
    clock.tick(60)
    screen.fill(BLACK)
    WIDTH = 32
    HEIGTH = 32
    # generazione della mappa
    map = map_generation(WIDTH,HEIGTH)
    # creazione del labirinto
    start = time.time()
    map = map_routing(map,WIDTH,HEIGTH)
    drawGrid(WIDTH,HEIGTH,map)
    break

python version:

Python 3.11.5

codon version:
0.16.3

numpy version:

1.24.4

Thank you in advance for the help ❤️