imrpoved del_node func(4)

Author: ShauryaDusht

data_structures/binary_tree/avl_tree.py

@@ -10,7 +10,7 @@
 
 import math
 import random
-from typing import Any
+from typing import Any, Optional
 
 
 class MyQueue:
@@ -208,7 +208,7 @@ def get_min_value_node(node: MyNode) -> MyNode:
     return current
 
 
-def del_node(root: MyNode, data: Any) -> MyNode | None:
+def del_node(root: Optional[MyNode], data: Any) -> Optional[MyNode]:
     if root is None:
         print("Nothing to delete")
         return None
@@ -358,4 +358,4 @@ def _test() -> None:
     random.shuffle(lst)
     for i in lst:
         t.del_node(i)
-        print(str(t))
+        print(str(t))