From ddd2755ac26e37e72d7fb7b8adce98f35651870d Mon Sep 17 00:00:00 2001 From: Mike Knepper Date: Thu, 27 Jul 2023 16:38:11 -0500 Subject: [PATCH] Drop relational drawing spike (#143) --- src/gretel_trainer/relational/drawing.py | 11 ----------- 1 file changed, 11 deletions(-) delete mode 100644 src/gretel_trainer/relational/drawing.py diff --git a/src/gretel_trainer/relational/drawing.py b/src/gretel_trainer/relational/drawing.py deleted file mode 100644 index f8768680..00000000 --- a/src/gretel_trainer/relational/drawing.py +++ /dev/null @@ -1,11 +0,0 @@ -import matplotlib.pyplot as plt -import networkx as nx -from networkx.drawing.nx_agraph import graphviz_layout - -from gretel_trainer.relational.core import RelationalData - - -def draw(rel_data: RelationalData) -> None: - pos = graphviz_layout(rel_data.graph, prog="dot") - nx.draw(rel_data.graph, pos, with_labels=True, arrows=True) - plt.show()