Rename client to TrainerClient()

andreyvelich · google-oss-robot · commit e9d2ba0baa39 · 2026-03-16T14:41:26.000Z
Signed-off-by: Andrey Velichkevich &lt;andrey.velichkevich@gmail.com&gt;
diff --git a/examples/deepspeed/text-summarization/T5-Fine-Tuning.ipynb b/examples/deepspeed/text-summarization/T5-Fine-Tuning.ipynb
@@ -834,7 +834,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "client.wait_for_job_status(name=job_id, timeout=20)"
+    "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"
    ]
   },
   {
diff --git a/examples/mlx/image-classification/mnist.ipynb b/examples/mlx/image-classification/mnist.ipynb
@@ -772,7 +772,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "client.wait_for_job_status(name=job_id, timeout=20)"
+    "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"
    ]
   },
   {
diff --git a/examples/mlx/language-modeling/fine-tune-llama.ipynb b/examples/mlx/language-modeling/fine-tune-llama.ipynb
@@ -597,7 +597,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "client.wait_for_job_status(name=job_id, timeout=20)"
+    "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"
    ]
   },
   {
diff --git a/examples/pytorch/question-answering/fine-tune-distilbert.ipynb b/examples/pytorch/question-answering/fine-tune-distilbert.ipynb
@@ -489,7 +489,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "client.wait_for_job_status(name=job_id, timeout=20)"
+    "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -834,7 +834,7 @@`
`834`	`834`	`"metadata": {},`
`835`	`835`	`"outputs": [],`
`836`	`836`	`"source": [`
`837`		`- "client.wait_for_job_status(name=job_id, timeout=20)"`
	`837`	`+ "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"`
`838`	`838`	`]`
`839`	`839`	`},`
`840`	`840`	`{`
Original file line number	Diff line number	Diff line change
`@@ -772,7 +772,7 @@`
`772`	`772`	`"metadata": {},`
`773`	`773`	`"outputs": [],`
`774`	`774`	`"source": [`
`775`		`- "client.wait_for_job_status(name=job_id, timeout=20)"`
	`775`	`+ "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"`
`776`	`776`	`]`
`777`	`777`	`},`
`778`	`778`	`{`
Original file line number	Diff line number	Diff line change
`@@ -597,7 +597,7 @@`
`597`	`597`	`"metadata": {},`
`598`	`598`	`"outputs": [],`
`599`	`599`	`"source": [`
`600`		`- "client.wait_for_job_status(name=job_id, timeout=20)"`
	`600`	`+ "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"`
`601`	`601`	`]`
`602`	`602`	`},`
`603`	`603`	`{`
Original file line number	Diff line number	Diff line change
`@@ -489,7 +489,7 @@`
`489`	`489`	`"metadata": {},`
`490`	`490`	`"outputs": [],`
`491`	`491`	`"source": [`
`492`		`- "client.wait_for_job_status(name=job_id, timeout=20)"`
	`492`	`+ "TrainerClient().wait_for_job_status(name=job_id, timeout=20)"`
`493`	`493`	`]`
`494`	`494`	`},`
`495`	`495`	`{`