utilities removed

DAGWorks-Inc · Mar 4, 2024 · 86736c0 · 86736c0
1 parent d4e9330
commit 86736c0
Showing 1 changed file with 0 additions and 5 deletions.
diff --git 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)