def METHOD_NAME(tmpdir):
    input_path = os.path.join(DATA_START, "criteo/crit_int_pq")
    output_path = os.path.join(DATA_START, "criteo/crit_test")
    _run_notebook(
        tmpdir,
        os.path.join(dirname(TEST_PATH), "torch", "criteo-example-petastorm.ipynb"),
        input_path,
        output_path,
        # disable rmm.reinitialize, seems to be causing issues
        batch_size=100000,
    )