From b31812cb9bbca38e1d0d2062fb4cf6b8d9968a1b Mon Sep 17 00:00:00 2001 From: Dustin Zelle Date: Thu, 13 Oct 2022 22:40:31 -0700 Subject: [PATCH] Update runner g3doc method signature. PiperOrigin-RevId: 481063085 --- tensorflow_gnn/docs/guide/runner.md | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tensorflow_gnn/docs/guide/runner.md b/tensorflow_gnn/docs/guide/runner.md index e51677d9..3bf77a5d 100644 --- a/tensorflow_gnn/docs/guide/runner.md +++ b/tensorflow_gnn/docs/guide/runner.md @@ -286,11 +286,14 @@ def run(*, gtspec: tfgnn.GraphTensorSpec, global_batch_size: int, epochs: int = 1, - drop_remainder: bool = True, + drop_remainder: bool = False, export_dirs: Optional[Sequence[str]] = None, + model_exporters: Optional[Sequence[ModelExporter]] = None, feature_processors: Optional[Sequence[GraphTensorProcessorFn]] = None, - valid_ds_provider: Optional[DatasetProvider] = None): - pass + valid_ds_provider: Optional[DatasetProvider] = None, + train_padding: Optional[GraphTensorPadding] = None, + valid_padding: Optional[GraphTensorPadding] = None, + tf_data_service_config: Optional[TFDataServiceConfig] = None): ``` The `model_fn` is expected to take a `tfgnn.GraphTensorSpec` and return a