def onco_tsg_count(driver_dataframe):
    """Function to count number of both onco and tumor suppressor genes from final driver gene dataframe."""
    onco_tsg = driver_dataframe["driver_role"] == "Oncogene/TSG"
    df_onco_tsg = driver_dataframe[onco_tsg][[
        "gene", "driver_role"]].drop_duplicates().reset_index(drop=True)
    return len(df_onco_tsg.index)