src/sage/doctest/parsing.py: Fix handling of '--probe all'

sagemath · Jan 28, 2024 · 1ada49c · 1ada49c
1 parent 3dd953c
commit 1ada49c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/sage/doctest/parsing.py b/src/sage/doctest/parsing.py
@@ -877,7 +877,7 @@ class SageDocTestParser(doctest.DocTestParser):
     optional_tags: Union[bool, set[str]]
     optional_only: bool
     optionals: dict[str, int]
-    probed_tags: set[str]
+    probed_tags: Union[bool, set[str]]
 
     def __init__(self, optional_tags=(), long=False, *, probed_tags=(), file_optional_tags=()):
         r"""
@@ -916,7 +916,10 @@ def __init__(self, optional_tags=(), long=False, *, probed_tags=(), file_optiona
                 self.optional_tags.remove('sage')
             else:
                 self.optional_only = True
-        self.probed_tags = set(probed_tags)
+        if probed_tags is True:
+            self.probed_tags = True
+        else:
+            self.probed_tags = set(probed_tags)
         self.file_optional_tags = set(file_optional_tags)
 
     def __eq__(self, other):