From bf6c7853c40f091ebb340a5e77095a0fd7211268 Mon Sep 17 00:00:00 2001
From: Oleksandr Ferludin <aferludin@google.com>
Date: Wed, 13 Dec 2023 08:49:07 -0800
Subject: [PATCH] Fixed sampler guide example for crediccard using DirectRunner

PiperOrigin-RevId: 590612793
---
 tensorflow_gnn/docs/guide/sampler.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow_gnn/docs/guide/sampler.md b/tensorflow_gnn/docs/guide/sampler.md
index c6674ae2..a2610114 100644
--- a/tensorflow_gnn/docs/guide/sampler.md
+++ b/tensorflow_gnn/docs/guide/sampler.md
@@ -128,6 +128,7 @@ Beam direct runner.
 ```
 cd <path-to>/gnn/examples/sampler/creditcard
 python3 -m tensorflow_gnn.experimental.sampler.beam.sampler \
+  --data_path="." \
   --graph_schema graph_schema.pbtxt \
   --sampling_spec sampling_spec.pbtxt \
   --output_samples outputs/examples.tfrecord \