fix(script)!: 'CN' is choosed as the cls_name

JamzumSum · Oct 3, 2022 · 533c122 · 533c122
1 parent af881c3
commit 533c122
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/test-pub.yml b/.github/workflows/test-pub.yml
@@ -3,7 +3,7 @@ name: Test, Build and Publish
 on:
   push:
     paths:
-      - "src/yacs_stubgen/**"
+      - "src/**"
       - "poetry.lock"
       - ".github/workflows/test-pub.yml"
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "yacs-stubgen"
-version = "0.2.0"
+version = "0.2.1"
 description = "Generate stub file for yacs config."
 authors = ["JamzumSum <zzzzss990315@gmail.com>"]
 license = "MIT"

diff --git a/src/genstub.py b/src/genstub.py
@@ -33,7 +33,7 @@ def inspect_module(mod_path: Path, ROOT: Path):
             cfg = v
             varname = k
         elif isinstance(v, type):
-            if issubclass(v, CfgNode):
+            if k != "CN" and issubclass(v, CfgNode):
                 clsname = k
         if clsname and varname:
             break