Allow users to use from pytask import mark. (#442)

Author: tobiasraabe

Diff for: docs/source/changes.md

@@ -56,6 +56,7 @@ releases are available on [PyPI](https://pypi.org/project/pytask) and
   {func}`pytask.is_task_function`.
 - {pull}`438` clarifies some types.
 - {pull}`440` refines more types.
+- {pull}`442` allows users to import `from pytask import mark` and use `@mark.skip`.
 
 ## 0.3.2 - 2023-06-07
 

Diff for: src/_pytask/collect.py

@@ -22,6 +22,7 @@
 from _pytask.console import get_file
 from _pytask.console import is_jupyter
 from _pytask.exceptions import CollectionError
+from _pytask.mark import MarkGenerator
 from _pytask.mark_utils import has_mark
 from _pytask.node_protocols import PNode
 from _pytask.node_protocols import PPathNode
@@ -183,6 +184,12 @@ def pytask_collect_file(
 
         collected_reports = []
         for name, obj in inspect.getmembers(mod):
+            # Skip mark generator since it overrides __getattr__ and seems like any
+            # object. Happens when people do ``from pytask import mark`` and
+            # ``@mark.x``.
+            if isinstance(obj, MarkGenerator):
+                continue
+
             # Ensures that tasks with this decorator are only collected once.
             if has_mark(obj, "task"):
                 continue

Diff for: src/_pytask/report.py

@@ -31,7 +31,6 @@ def from_exception(
         exc_info: OptionalExceptionInfo,
         node: MetaNode | None = None,
     ) -> CollectionReport:
-        exc_info = remove_internal_traceback_frames_from_exc_info(exc_info)
         return cls(outcome=outcome, node=node, exc_info=exc_info)
 
 

Diff for: tests/test_mark.py

@@ -395,3 +395,17 @@ def task_write_text(): ...
         check=False,
     )
     assert b"FutureWarning: '@pytask.mark.task'" in result.stdout
+
+
+@pytest.mark.end_to_end()
+def test_different_mark_import(runner, tmp_path):
+    source = """
+    from pytask import mark
+
+    @mark.skip
+    def task_write_text(): ...
+    """
+    tmp_path.joinpath("task_module.py").write_text(textwrap.dedent(source))
+    result = runner.invoke(cli, [tmp_path.as_posix()])
+    assert result.exit_code == ExitCode.OK
+    assert "Skipped" in result.output