diff --git a/pydra/engine/tests/test_audit.py b/pydra/engine/tests/test_audit.py index a4ac3c7ac..52f09f88b 100644 --- a/pydra/engine/tests/test_audit.py +++ b/pydra/engine/tests/test_audit.py @@ -302,3 +302,15 @@ def TestFunc(a: int, b: float = 0.1): # commented out to speed up testing collect_messages(tmpdir / funky._checksum, message_path, ld_op="compact") assert (tmpdir / funky._checksum / "messages.jsonld").exists() + + +def test_audit_resource(): + + @python.define(outputs=["out"]) + def AddVar(a: int, b: int) -> int: + return a + b + + task0 = AddVar(a=4, b=5) + outputs = task0(audit_flags=AuditFlag.RESOURCE) + + print(outputs) diff --git a/pyproject.toml b/pyproject.toml index e93f75dc9..2579b9f5e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,6 +14,7 @@ dependencies = [ "filelock >=3.0.0", "fileformats >=0.15.0a7", "platformdirs >=2", + "psutil >=5.9.0", ] license = { file = "LICENSE" } authors = [{ name = "Nipype developers", email = "neuroimaging@python.org" }] @@ -78,7 +79,6 @@ test = [ "fileformats-extras >=0.15.0a7", "numpy >=1.26", "pyld >=2.0", - "psutil >=5.9.0", "python-dateutil >=2.8.2", "tornado >=6.1", ]