diff --git a/src/graph_notebook/magics/ml.py b/src/graph_notebook/magics/ml.py index 320bcc59..7847271b 100644 --- a/src/graph_notebook/magics/ml.py +++ b/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: diff --git a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-01-Introduction-to-Node-Classification-Gremlin.ipynb b/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-01-Introduction-to-Node-Classification-Gremlin.ipynb index 8ca3ad99..5ec6d05c 100644 --- a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-01-Introduction-to-Node-Classification-Gremlin.ipynb +++ b/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": [], diff --git a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-02-Introduction-to-Node-Regression-Gremlin.ipynb b/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-02-Introduction-to-Node-Regression-Gremlin.ipynb index 5417b613..75e42f75 100644 --- a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-02-Introduction-to-Node-Regression-Gremlin.ipynb +++ b/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}\"\"\"" ] }, diff --git a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-03-Introduction-to-Link-Prediction-Gremlin.ipynb b/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-03-Introduction-to-Link-Prediction-Gremlin.ipynb index f50cf921..e77a06e9 100644 --- a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-03-Introduction-to-Link-Prediction-Gremlin.ipynb +++ b/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}\"\"\"" ] }, diff --git a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-04-Introduction-to-Edge-Classification-Gremlin.ipynb b/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-04-Introduction-to-Edge-Classification-Gremlin.ipynb index f2b1c6cb..accd383c 100644 --- a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-04-Introduction-to-Edge-Classification-Gremlin.ipynb +++ b/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}\"\"\"" ] }, diff --git a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-05-Introduction-to-Edge-Regression-Gremlin.ipynb b/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-05-Introduction-to-Edge-Regression-Gremlin.ipynb index 9977384d..3dc05098 100644 --- a/src/graph_notebook/notebooks/04-Machine-Learning/Neptune-ML-05-Introduction-to-Edge-Regression-Gremlin.ipynb +++ b/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}\"\"\"" ] },