From ecd7a55510929e80bae9de1f05aa82294b83c637 Mon Sep 17 00:00:00 2001
From: 0x00-pl <0x00.pl@gmail.com>
Date: Tue, 24 Sep 2024 02:09:01 +0800
Subject: [PATCH] typo

---
 plai/core/module.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/plai/core/module.py b/plai/core/module.py
index 67d1e29..5c7cb9f 100644
--- a/plai/core/module.py
+++ b/plai/core/module.py
@@ -38,7 +38,7 @@ def __init_subclass__(cls, **kwargs):
         Node.subclass_dict[op_name] = cls
 
     @staticmethod
-    def get_op_subclass(op_name: str):
+    def get_node_class(op_name: str):
         assert op_name in Node.subclass_dict, f'Unregister Class with name: {op_name}'
         return Node.subclass_dict[op_name]