sdk tests with papermill

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

Author: Yehudit Kerido

.github/workflows/sdk-e2e-tests-with-papermill.yaml .github/workflows/e2e-test-notebooks.yaml

@@ -96,6 +96,7 @@
    "execution_count": 1,
    "id": "98a77484-cee4-4d77-bd76-915c469a242b",
    "metadata": {
+    "editable": true,
     "execution": {
      "iopub.execute_input": "2022-09-12T18:46:59.051290Z",
      "iopub.status.busy": "2022-09-12T18:46:59.050544Z",
@@ -106,6 +107,9 @@
     "pycharm": {
      "name": "#%%\n"
     },
+    "slideshow": {
+     "slide_type": ""
+    },
     "tags": []
    },
    "outputs": [],
@@ -549,17 +553,14 @@
     "\n",
     "# Start TFJob Training.\n",
     "tfjob_name = \"train-mnist\"\n",
-    "tfjob_client = TrainingClient(namespace=namespace)\n",
+    "tfjob_client = TrainingClient(namespace=namespace, namespace=namespace, job_kind=\"TFJob\")\n",
     "\n",
     "#create_tfjob_from_func\n",
     "tfjob_client.create_job(\n",
     "    name=tfjob_name,\n",
-    "    namespace=namespace,\n",
-    "    job_kind=\"TFJob\",\n",
     "    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",
     ")"
    ]
   },
@@ -608,7 +609,7 @@
     }
    ],
    "source": [
-    "print(f\"TFJob status: {tfjob_client.get_job_conditions(tfjob_name, namespace=namespace, job_kind='TFJob')}\")"
+    "print(f\"TFJob status: {tfjob_client.get_job_conditions(tfjob_name)}\")"
    ]
   },
   {
@@ -764,7 +765,7 @@
     }
    ],
    "source": [
-    "tfjob_client.get_job_logs(name=tfjob_name, namespace=namespace, is_master=True, follow=True, job_kind=\"TFJob\")"
+    "tfjob_client.get_job_logs(name=tfjob_name, is_master=True, follow=True)"
    ]
   },
   {
@@ -849,7 +850,7 @@
     }
    ],
    "source": [
-    "tfjob_client.delete_job(tfjob_name, namespace=namespace, job_kind=\"TFJob\")"
+    "tfjob_client.delete_job(tfjob_name)"
    ]
   },
   {