diff options
Diffstat (limited to 'coverage/disposition.py')
-rw-r--r-- | coverage/disposition.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/coverage/disposition.py b/coverage/disposition.py index 5237c364..3cc6c8d6 100644 --- a/coverage/disposition.py +++ b/coverage/disposition.py @@ -7,6 +7,8 @@ from __future__ import annotations from typing import Optional, Type, TYPE_CHECKING +from coverage.types import TFileDisposition + if TYPE_CHECKING: from coverage.plugin import FileTracer @@ -30,7 +32,7 @@ class FileDisposition: # be implemented in either C or Python. Acting on them is done with these # functions. -def disposition_init(cls: Type[FileDisposition], original_filename: str) -> FileDisposition: +def disposition_init(cls: Type[TFileDisposition], original_filename: str) -> TFileDisposition: """Construct and initialize a new FileDisposition object.""" disp = cls() disp.original_filename = original_filename @@ -43,7 +45,7 @@ def disposition_init(cls: Type[FileDisposition], original_filename: str) -> File return disp -def disposition_debug_msg(disp: FileDisposition) -> str: +def disposition_debug_msg(disp: TFileDisposition) -> str: """Make a nice debug message of what the FileDisposition is doing.""" if disp.trace: msg = f"Tracing {disp.original_filename!r}" |