Fix class() in BFS and DFS

hansonchar · Jan 14, 2025 · 2dfec22 · 2dfec22
1 parent 6ef3690
commit 2dfec22
Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/learning-lua/algo/BFS.lua b/learning-lua/algo/BFS.lua
@@ -26,8 +26,8 @@ function BFS:iterate()
 end
 
 function BFS:class(o)
-  setmetatable(o, BFS)
-  BFS.__index = BFS
+  setmetatable(o, self)
+  self.__index = self
   return o
 end
 

diff --git a/learning-lua/algo/DFS.lua b/learning-lua/algo/DFS.lua
@@ -41,8 +41,8 @@ function DFS:iterate()
 end
 
 function DFS:class(o)
-  setmetatable(o, DFS)
-  DFS.__index = DFS
+  setmetatable(o, self)
+  self.__index = self
   return o
 end