Generate namespace semantic tokens for submodules

jbradaric · Aug 31, 2023 · 0a09d34 · 0a09d34
1 parent 8f2e392
commit 0a09d34
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/pyright-internal/src/analyzer/semanticTokensWalker.ts b/packages/pyright-internal/src/analyzer/semanticTokensWalker.ts
@@ -84,6 +84,10 @@ export class SemanticTokensWalker extends ParseTreeWalker {
                     this._addItem(node.memberName.start, node.memberName.length, SemanticTokenTypes.class, []);
                     break;
                 }
+
+            case TypeCategory.Module:
+                this._addItem(node.memberName.start, node.memberName.length, SemanticTokenTypes.namespace, []);
+                break;
             // fallthrough
 
             default: