diff --git a/tests/test_driver_export.py b/tests/test_driver_export.py index 44a7c47fd..07f1fdcad 100644 --- a/tests/test_driver_export.py +++ b/tests/test_driver_export.py @@ -96,13 +96,8 @@ ], } -from hamilton import graph_types - def test_export_execution(): dr = driver.Builder().with_modules(no_parallel).build() - g = graph_types.HamiltonGraph.from_graph(dr.graph) - for n in g.nodes: - print(n.name, n.version) json_str = dr.export_execution(["final"]) assert json_str == json.dumps(EXPECTED_JSON)