diff --git a/tests/integration_tests/test_single_node.py b/tests/integration_tests/test_single_node.py index 796bd7d6..c3e257b4 100644 --- a/tests/integration_tests/test_single_node.py +++ b/tests/integration_tests/test_single_node.py @@ -66,7 +66,7 @@ def run(self): def test_run(proj_path): test_node_1 = ExampleNode01(inputs="Lorem Ipsum") assert test_node_1.inputs == "Lorem Ipsum" - test_node_1.write_graph() + test_node_1.write_graph(write_desc=True) obj = ExampleNode01.load() assert obj.inputs == "Lorem Ipsum" diff --git a/zntrack/core/base.py b/zntrack/core/base.py index d4295276..6e303283 100644 --- a/zntrack/core/base.py +++ b/zntrack/core/base.py @@ -498,7 +498,7 @@ def write_graph( # noqa: C901 no_run_cache: bool = False, dry_run: bool = False, run: bool = None, - write_desc: bool = True, + write_desc: bool = False, *, call_args: str = None, ):