Correct naming of ID parameter for NeptuneML Endpoint command (#217)

* Correct naming of ID parameter for NeptuneML endpoint command

* Update endpoint_status ID help

Co-authored-by: Michael Chin <[email protected]>

Author: michaelnchin

src/graph_notebook/magics/ml.py

@@ -279,7 +279,7 @@ def generate_neptune_ml_parser():
     endpoint_subparsers = parser_endpoint.add_subparsers(help='endpoint sub-command help',
                                                          dest='which_sub')
     endpoint_start_parser = endpoint_subparsers.add_parser('create', help='create a new endpoint')
-    endpoint_start_parser.add_argument('--job-id', type=str, default='')
+    endpoint_start_parser.add_argument('--id', type=str, default='A unique identifier for the new inference endpoint.')
     endpoint_start_parser.add_argument('--model-training-job-id', type=str, default='',
                                        help='The job Id of the completed model-training job. '
                                             'You must supply either model-training-job-id or model-transform-job-id.')
@@ -313,7 +313,7 @@ def generate_neptune_ml_parser():
     endpoint_status_parser = endpoint_subparsers.add_parser('status',
                                                             help='obtain the status of an existing endpoint '
                                                                  'creation job')
-    endpoint_status_parser.add_argument('--job-id', type=str, default='')
+    endpoint_status_parser.add_argument('--id', type=str, default='The ID of an existing inference endpoint.')
     endpoint_status_parser.add_argument('--store-to', type=str, default='', help='store result to this variable')
     endpoint_status_parser.add_argument('--wait', action='store_true',
                                         help='wait for the exporter to finish running')
@@ -635,7 +635,7 @@ def neptune_ml_endpoint(args: argparse.Namespace, client: Client, output: widget
     if args.which_sub == 'create':
         if params is None or params == '' or params == {}:
             params = {
-                "id": args.job_id,
+                "id": args.id,
                 'instanceType': args.instance_type
             }
             if args.update:
@@ -699,9 +699,9 @@ def neptune_ml_endpoint(args: argparse.Namespace, client: Client, output: widget
             return create_endpoint_job
     elif args.which_sub == 'status':
         if args.wait:
-            return wait_for_endpoint(args.job_id, client, output, args.wait_interval, args.wait_timeout)
+            return wait_for_endpoint(args.id, client, output, args.wait_interval, args.wait_timeout)
         else:
-            endpoint_status = client.endpoints_status(args.job_id)
+            endpoint_status = client.endpoints_status(args.id)
             endpoint_status.raise_for_status()
             return endpoint_status.json()
     else:

src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-01-Introduction-to-Node-Classification-Gremlin.ipynb

@@ -480,7 +480,7 @@
    "execution_count": null,
    "source": [
     "endpoint_params=f\"\"\"\n",
-    "--job-id {training_job_name} \n",
+    "--id {training_job_name}\n",
     "--model--training-job-id {training_job_name} \"\"\""
    ],
    "outputs": [],

src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-02-Introduction-to-Node-Regression-Gremlin.ipynb

@@ -476,7 +476,7 @@
    "outputs": [],
    "source": [
     "endpoint_params=f\"\"\"\n",
-    "--job-id {training_job_name} \n",
+    "--id {training_job_name}\n",
     "--model-training-job-id {training_job_name}\"\"\""
    ]
   },

src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-03-Introduction-to-Link-Prediction-Gremlin.ipynb

@@ -450,7 +450,7 @@
    "outputs": [],
    "source": [
     "endpoint_params=f\"\"\"\n",
-    "--job-id {training_job_name} \n",
+    "--id {training_job_name}\n",
     "--model-training-job-id {training_job_name}\"\"\""
    ]
   },

src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-04-Introduction-to-Edge-Classification-Gremlin.ipynb

@@ -479,7 +479,7 @@
    "outputs": [],
    "source": [
     "endpoint_params=f\"\"\"\n",
-    "--job-id {training_job_name} \n",
+    "--id {training_job_name}\n",
     "--model-training-job-id {training_job_name}\"\"\""
    ]
   },

src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-05-Introduction-to-Edge-Regression-Gremlin.ipynb

@@ -477,7 +477,7 @@
    "outputs": [],
    "source": [
     "endpoint_params=f\"\"\"\n",
-    "--job-id {training_job_name} \n",
+    "--id {training_job_name}\n",
     "--model-training-job-id {training_job_name}\"\"\""
    ]
   },