Fix types in Memory

Author: mdbrnowski

utils/memory.py

@@ -1,30 +1,31 @@
 from .values import Value
+from .types import Type
 
 
 class Memory:
     def __init__(self):
-        self.variables: dict[str, Value] = {}
+        self.variables: dict[str, Value | Type] = {}
 
     def has_variable(self, name: str) -> bool:
         return name in self.variables
 
-    def get(self, name: str) -> Value:
+    def get(self, name: str) -> Value | Type:
         return self.variables[name]
 
-    def put(self, name: str, value: Value):
+    def put(self, name: str, value: Value | Type):
         self.variables[name] = value
 
 
 class MemoryStack:
     def __init__(self):
         self.stack: list[Memory] = []
 
-    def get(self, name: str) -> Value:
+    def get(self, name: str) -> Value | Type:
         for memory in self.stack:
             if memory.has_variable(name):
                 return memory.get(name)
 
-    def put(self, name: str, value: Value):
+    def put(self, name: str, value: Value | Type):
         for memory in self.stack:
             if memory.has_variable(name):
                 memory.put(name, value)