From 557c7ce0bfae5087d1f15914fa827b3f0a942de4 Mon Sep 17 00:00:00 2001 From: Yusuke Kadowaki Date: Sun, 30 Oct 2022 01:06:13 +0900 Subject: [PATCH] Fix Literal for older version --- src/_pytest/tmpdir.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/_pytest/tmpdir.py b/src/_pytest/tmpdir.py index 52355dff7..ebdb7fe5e 100644 --- a/src/_pytest/tmpdir.py +++ b/src/_pytest/tmpdir.py @@ -13,6 +13,8 @@ from typing import Union if TYPE_CHECKING: from typing_extensions import Literal + RetentionType = Literal["all", "failed", "none"] + import attr from _pytest.config.argparsing import Parser @@ -45,13 +47,13 @@ class TempPathFactory: _trace = attr.ib() _basetemp = attr.ib(type=Optional[Path]) _retention_count = attr.ib(type=int) - _retention_policy = attr.ib(type=RetentionPolicy) + _retention_policy = attr.ib(type="RetentionType") def __init__( self, given_basetemp: Optional[Path], retention_count: int, - retention_policy: RetentionPolicy, + retention_policy: "RetentionType", trace, basetemp: Optional[Path] = None, *,