def METHOD_NAME():
    modified_notebook_path = os.path.join(local_notebooks_dir, "exercise_2_test.ipynb")
    nb = nbformat.read(
        os.path.join(local_notebooks_dir, "exercise_2.ipynb"),
        as_version=nbformat.NO_CONVERT,
    )
    _replace_str(
        nb,
        'path = "s3://dask-data/nyc-taxi/2015/yellow_tripdata_2015-01.csv"',
        '# path = "s3://dask-data/nyc-taxi/2015/yellow_tripdata_2015-01.csv"',
    )
    new_optional_cell = f'path = "{test_dataset_path}"\n' + download_taxi_dataset
    optional_cell_idx = _find_code_cell_idx(nb, "[Optional] Download data locally.")
    nb["cells"][optional_cell_idx]["source"] = new_optional_cell
    nbformat.write(nb, modified_notebook_path)
    _execute_notebook(modified_notebook_path)