Wait for element to disappear/ not be preset anymore

I use python client for very similar use case and I am able to do it successfully with following code

def wait_until_element_is_invisible(driver, find_by, wait_until=10):
return WebDriverWait(driver, wait_until).until(EC.invisibility_of_element_located(find_by))