Value -> Scalar

1 files changed, 3 insertions, 3 deletions

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