Skip to content

Commit c887858

Browse files
authored
Merge pull request #433 from opsmill/pog-transform-node-argument
Add transform infrahub_node argument
2 parents 8f4a678 + 1018811 commit c887858

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

infrahub_sdk/pytest_plugin/items/python_transform.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import ujson
88
from httpx import HTTPStatusError
99

10+
from ...node import InfrahubNode
1011
from ..exceptions import OutputMatchError, PythonTransformDefinitionError
1112
from ..models import InfrahubTestExpectedResult
1213
from .base import InfrahubItem
@@ -41,7 +42,7 @@ def instantiate_transform(self) -> None:
4142
)
4243
client = self.session.infrahub_client # type: ignore[attr-defined]
4344
# TODO: Look into seeing how a transform class may use the branch, but set as a empty string for the time being to keep current behaviour
44-
self.transform_instance = transform_class(branch="", client=client)
45+
self.transform_instance = transform_class(branch="", client=client, infrahub_node=InfrahubNode)
4546

4647
def run_transform(self, variables: dict[str, Any]) -> Any:
4748
self.instantiate_transform()

0 commit comments

Comments
 (0)