Fix.

Author: tobiasraabe

pyproject.toml

@@ -15,7 +15,7 @@ classifiers = [
     "Programming Language :: Python :: 3 :: Only",
 ]
 requires-python = ">=3.8"
-dependencies = ["click", "pytask>=0.3,<0.4"]
+dependencies = ["click", "pytask>=0.4"]
 dynamic = ["version"]
 
 [project.readme]
@@ -38,6 +38,9 @@ pytask = { pytask_stata = "pytask_stata.plugin" }
 
 [tool.rye]
 managed = true
+dev-dependencies = [
+    "tox-uv>=1.8.2",
+]
 
 [tool.hatch.build.targets.sdist]
 exclude = ["tests"]

src/pytask_stata/collect.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import functools
 import subprocess
 import warnings
 from pathlib import Path
@@ -121,7 +122,7 @@ def pytask_collect_task(
             node_info=NodeInfo(
                 arg_name="_cwd",
                 path=(),
-                value=path.parent,
+                value=path.parent.as_posix(),
                 task_path=path,
                 task_name=name,
             ),
@@ -140,13 +141,14 @@ def pytask_collect_task(
         dependencies["_cwd"] = cwd_node
         dependencies["_executable"] = executable_node
 
+        partialed = functools.partial(run_stata_script, _cwd=path.parent)
         markers = obj.pytask_meta.markers if hasattr(obj, "pytask_meta") else []
 
         task: PTask
         if path is None:
             task = TaskWithoutPath(
                 name=name,
-                function=run_stata_script,
+                function=partialed,
                 depends_on=dependencies,
                 produces=products,
                 markers=markers,
@@ -155,7 +157,7 @@ def pytask_collect_task(
             task = Task(
                 base_name=name,
                 path=path,
-                function=run_stata_script,
+                function=partialed,
                 depends_on=dependencies,
                 produces=products,
                 markers=markers,

src/pytask_stata/shared.py

@@ -46,7 +46,7 @@
 
 def stata(
     *,
-    script: str | Path | None = None,
+    script: str | Path,
     options: str | Iterable[str] | None = None,
 ) -> tuple[str | Path | None, str | Iterable[str] | None]:
     """Specify command line options for Stata.

tests/conftest.py

@@ -7,6 +7,7 @@
 
 import pytest
 from click.testing import CliRunner
+from pytask import storage
 from pytask_stata.config import STATA_COMMANDS
 
 needs_stata = pytest.mark.skipif(
@@ -72,6 +73,7 @@ def _restore_sys_path_and_module_after_test_execution():
 class CustomCliRunner(CliRunner):
     def invoke(self, *args, **kwargs):
         """Restore sys.path and sys.modules after an invocation."""
+        storage.create()
         with restore_sys_path_and_module_after_test_execution():
             return super().invoke(*args, **kwargs)
 

tests/test_execute.py

@@ -105,10 +105,10 @@ def run_do_file():
 @pytest.mark.end_to_end()
 def test_raise_error_if_stata_is_not_found(tmp_path, monkeypatch):
     task_source = """
-    import pytask
+    from pytask import mark, task
 
-    @pytask.mark.stata(script="script.do")
-    @pytask.mark.produces("out.dta")
+    @task(kwargs={"produces": "out.dta"})
+    @mark.stata(script="script.do")
     def task_run_do_file():
         pass
     """
@@ -150,6 +150,7 @@ def task_run_do_file():
     result = runner.invoke(cli, [tmp_path.as_posix()])
 
     assert result.exit_code == ExitCode.OK
+    assert tmp_path.joinpath("out.dta").exists()
 
 
 @needs_stata

tests/test_normal_execution_w_plugin.py

@@ -5,6 +5,7 @@
 import textwrap
 
 import pytest
+from pytask import ExitCode
 from pytask import cli
 
 
@@ -36,4 +37,4 @@ def task_example(depends_on, produces):
         tmp_path.joinpath(dependency).touch()
 
     result = runner.invoke(cli, [tmp_path.as_posix()])
-    assert result.exit_code == 0
+    assert result.exit_code == ExitCode.OK