Python - Pasar instancia webdriver selenium por parametro a funcion en python

Holas

Estoy empezando con python y tengo la duda como puedo pasar una instancia de un webdriver de selenium por parametro a una funcion en python, con el fin de que este lenguaje me reconozca el tipo de objeto y asi me sugiera cierta funciones con el autocompletado.

En java se me hacia mas facil, ya que en los argumentos de las funciones definia el tipo de variable.

//En java

public boolean login(Webdriver driver, String username, String password) {

driver.click();
}

//En python no se como forzar o pedir un elemento tipo webdriver, para que me vayaa sugiriendo funciones.

def login(driver):

driver