Skip to content

Commit b31812c

Browse files
dzelletensorflower-gardener
authored andcommitted
Update runner g3doc method signature.
PiperOrigin-RevId: 481063085
1 parent e8ff8df commit b31812c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tensorflow_gnn/docs/guide/runner.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -286,11 +286,14 @@ def run(*,
286286
gtspec: tfgnn.GraphTensorSpec,
287287
global_batch_size: int,
288288
epochs: int = 1,
289-
drop_remainder: bool = True,
289+
drop_remainder: bool = False,
290290
export_dirs: Optional[Sequence[str]] = None,
291+
model_exporters: Optional[Sequence[ModelExporter]] = None,
291292
feature_processors: Optional[Sequence[GraphTensorProcessorFn]] = None,
292-
valid_ds_provider: Optional[DatasetProvider] = None):
293-
pass
293+
valid_ds_provider: Optional[DatasetProvider] = None,
294+
train_padding: Optional[GraphTensorPadding] = None,
295+
valid_padding: Optional[GraphTensorPadding] = None,
296+
tf_data_service_config: Optional[TFDataServiceConfig] = None):
294297
```
295298

296299
The `model_fn` is expected to take a `tfgnn.GraphTensorSpec` and return a

0 commit comments

Comments
 (0)