fix: NoneType ImportError on python<3.10

JamzumSum · Nov 5, 2022 · 8523dc1 · 8523dc1
1 parent cfd96dd
commit 8523dc1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/yacs_stubgen/__init__.py b/src/yacs_stubgen/__init__.py
@@ -1,5 +1,4 @@
 from pathlib import Path
-from types import NoneType
 from typing import Any, Dict, Optional, Union
 
 import yaml
@@ -8,7 +7,7 @@
 typo_map = {
     list: "T.Sequence",
     tuple: "T.Sequence",
-    NoneType: "T.Any",
+    type(None): "T.Any",
 }