Skip to content

Commit 6b2a639

Browse files
committed
Add smaller tweaks & improvements
1 parent 1825ba1 commit 6b2a639

3 files changed

Lines changed: 5 additions & 3 deletions

File tree

src/docstub/_analysis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def __post_init__(self):
155155
):
156156
raise ValueError("builtin cannot contain import information")
157157
elif self.import_name is None:
158-
raise ValueError("non bultin must at least define an `import_name`")
158+
raise ValueError("non builtin must at least define an `import_name`")
159159

160160
def __repr__(self):
161161
if self.builtin_name:

src/docstub/_docstrings.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ class Annotation:
4545

4646
def __post_init__(self):
4747
object.__setattr__(self, "imports", frozenset(self.imports))
48+
if "~" in self.value:
49+
raise ValueError(f"unexpected '~' in annotation value: {self.value}")
4850

4951
def __str__(self) -> str:
5052
return self.value

src/docstub/_stubs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def try_format_stub(stub: str) -> str:
149149
return stub
150150

151151

152-
class FuncType(enum.Enum):
152+
class FuncType(enum.StrEnum):
153153
MODULE = enum.auto()
154154
CLASS = enum.auto()
155155
FUNC = enum.auto()
@@ -526,7 +526,7 @@ def _parse_imports(imports, *, current_module=None):
526526
527527
Parameters
528528
----------
529-
imports : set[~.DocName]
529+
imports : set[~.KnownImport]
530530
current_module : str, optional
531531
532532
Returns

0 commit comments

Comments
 (0)