Fix review comments

Signed-off-by: Yehudit Kerido <[email protected]>

examples/v1beta1/sdk/tune-train-from-func.ipynb

@@ -441,6 +441,8 @@
     }
    ],
    "source": [
+    "import time\n",
+    "time.sleep(120)\n",
     "status = katib_client.is_experiment_succeeded(exp_name, namespace=namespace)\n",
     "print(f\"Katib Experiment is Succeeded: {status}\\n\")\n",
     "\n",
@@ -511,7 +513,7 @@
     "\n",
     "# Start TFJob Training.\n",
     "tfjob_name = \"train-mnist\"\n",
-    "tfjob_client = TrainingClient()\n",
+    "tfjob_client = TrainingClient(namespace=namespace)\n",
     "\n",
     "#create_tfjob_from_func\n",
     "tfjob_client.create_job(\n",
@@ -521,6 +523,7 @@
     "    train_func=train_mnist_model,\n",
     "    parameters=parameters, # Input parameters for the train function.\n",
     "    num_workers=5, # How many TFJob Workers will be run.\n",
+    "    base_image=\"tensorflow/tensorflow:2.10.0\",  # Use TensorFlow image\n",
     ")"
    ]
   },