From 22f6c7e5b3fd45480f2ef7827474fdb9d6c6f82c Mon Sep 17 00:00:00 2001 From: Raghuram Subramani Date: Wed, 29 May 2024 18:11:05 +0530 Subject: Value -> Scalar --- src/graph.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/graph.py') diff --git a/src/graph.py b/src/graph.py index 2067071..9aecde1 100644 --- a/src/graph.py +++ b/src/graph.py @@ -4,10 +4,10 @@ from graphviz import Digraph import matplotlib.pyplot as plt import matplotlib.image as mpimg -from .value import Value +from .scalar import Scalar class Graph: - def __init__(self, root: Value) -> None: + def __init__(self, root: Scalar) -> None: self.dot = Digraph(format='png', graph_attr={ 'rankdir': 'LR' }) self.nodes = set() @@ -16,7 +16,7 @@ class Graph: self.build(root) self.draw() - def build(self, x: Value): + def build(self, x: Scalar): self.nodes.add(x) for child in x._prev: # Add a line from child to x -- cgit v1.2.3