typo

0x00-pl · Jan 3, 2025 · ecd7a55 · ecd7a55
1 parent 89af118
commit ecd7a55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]