name 'nosuchelementexception' is not defined code example

Example: python exception element not found

from selenium.common.exceptions import NoSuchElementException

try:
    if driver.find_element_by_class_name(test_element_class_name).is_displayed():
        print('found')
except NoSuchElementException:
    pass