[mypy] Fix type annotations in data_structures/binary_tree (#5518)

* fix: fix mypy errors

Update binary_search_tree `arr` argument to be typed as a list
within `find_kth_smallest` function

Update return type of `merge_two_binary_trees` as both inputs can
be None which means that a None type value can be returned from
this function

* updating DIRECTORY.md

Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
This commit is contained in:
Sherman Hui
2021-10-22 07:07:05 -07:00
committed by GitHub
parent d82cf5292f
commit 629848e372
3 changed files with 5 additions and 2 deletions

View File

@ -151,7 +151,7 @@ class BinarySearchTree:
def find_kth_smallest(self, k: int, node: Node) -> int:
"""Return the kth smallest element in a binary search tree"""
arr = []
arr: list = []
self.inorder(arr, node) # append all values to list using inorder traversal
return arr[k - 1]