summaryrefslogtreecommitdiff
path: root/coverage/disposition.py
diff options
context:
space:
mode:
Diffstat (limited to 'coverage/disposition.py')
-rw-r--r--coverage/disposition.py6
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}"