Update runner g3doc method signature.

PiperOrigin-RevId: 481063085
tensorflow · Oct 14, 2022 · b31812c · b31812c
1 parent e8ff8df
commit b31812c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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