def METHOD_NAME(setup_cluster):
    sess = setup_cluster
    path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tf_dataset.py")
    data = mt.random.rand(1000, 32, dtype="f4")
    labels = mt.random.randint(0, 2, (1000, 10), dtype="f4")
    assert (
        run_tensorflow_script(
            path,
            n_workers=2,
            data={"feature_data": data, "labels": labels},
            command_argv=["multiple"],
            session=sess,
        ).fetch()["status"]
        == "ok"
    )