fixing the DFS and modify the main

a_maze_ing.py

@@ -7,8 +7,7 @@ import src.amaz_lib as g
 def main(maze_gen: MazeGenerator) -> None:
     # try:
     maze = Maze(maze=None)
-    generator = Kruskal()
+    for alg in maze_gen.generator(30, 10):
-    for alg in generator.generator(50, 50):
         maze.set_maze(alg)
         os.system("clear")
         maze.ascii_print()

src/amaz_lib/MazeGenerator.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from dataclasses import dataclass
 from typing import Generator, Set
 import numpy as np
 from .Cell import Cell
@@ -107,7 +106,7 @@ class Kruskal(MazeGenerator):
 class DepthFirstSearch(MazeGenerator):
 
     def generator(
-        self, width: int, height: int
+        self, height: int, width: int
     ) -> Generator[np.ndarray, None, np.ndarray]:
         maze = DepthFirstSearch.init_maze(width, height)
         visited = np.zeros((height, width), dtype=bool)