Skip to content

Commit 7672086

Browse files
Update docs/proposals/107-spark-client/README.md
Co-authored-by: Andrey Velichkevich <[email protected]> Signed-off-by: Shekhar Prasad Rajak <[email protected]>
1 parent db6f3b3 commit 7672086

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

docs/proposals/107-spark-client/README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -941,10 +941,11 @@ spark_client.wait_for_job_status(features.submission_id)
941941
# Step 2: Model training with Trainer
942942
trainer_client = TrainerClient()
943943
trainer_client.train(
944-
name="train-model",
945-
func=train_func,
946-
num_nodes=4,
947-
)
944+
trainer=CustomTrainer(
945+
func=train_func,
946+
num_nodes=4
947+
)
948+
)
948949
```
949950

950951
#### Integration with Pipelines

0 commit comments

Comments
 (0)