@@ -3,15 +3,15 @@ metadata:
3
3
annotations :
{author: Alexey Volkov <[email protected] >, canonical_location: 'https://raw.githubusercontent.com/Ark-kun/pipeline_components/master/components/google-cloud/Vertex_AI/Models/Upload_Tensorflow_model/component.yaml'}
4
4
inputs :
5
5
- {name: model, type: TensorflowSavedModel}
6
- - {name: tensorflow_version, type: String, default: '2.7', optional: true}
6
+ - {name: tensorflow_version, type: String, optional: true}
7
7
- name : use_gpu
8
8
type : Boolean
9
9
default : " False"
10
10
optional : true
11
- - {name: display_name, type: String, default: Tensorflow model, optional: true}
11
+ - {name: display_name, type: String, optional: true}
12
12
- {name: description, type: String, optional: true}
13
13
- {name: project, type: String, optional: true}
14
- - {name: location, type: String, default: us-central1, optional: true}
14
+ - {name: location, type: String, optional: true}
15
15
- {name: labels, type: JsonObject, optional: true}
16
16
- {name: staging_bucket, type: String, optional: true}
17
17
outputs :
@@ -24,9 +24,8 @@ implementation:
24
24
- sh
25
25
- -c
26
26
- (PIP_DISABLE_PIP_VERSION_CHECK=1 python3 -m pip install --quiet --no-warn-script-location
27
- ' git+https://github.com/Ark-kun/python-aiplatform@1e1cbfef76b7c5c8db705d5c4c17c3691de7b032#egg=google-cloud-aiplatform&subdirectory=.'
28
- || PIP_DISABLE_PIP_VERSION_CHECK=1 python3 -m pip install --quiet --no-warn-script-location
29
- ' git+https://github.com/Ark-kun/python-aiplatform@1e1cbfef76b7c5c8db705d5c4c17c3691de7b032#egg=google-cloud-aiplatform&subdirectory=.'
27
+ ' google-cloud-aiplatform==1.16.0' || PIP_DISABLE_PIP_VERSION_CHECK=1 python3
28
+ -m pip install --quiet --no-warn-script-location 'google-cloud-aiplatform==1.16.0'
30
29
--user) && "$0" "$@"
31
30
- sh
32
31
- -ec
@@ -37,10 +36,10 @@ implementation:
37
36
- |
38
37
def upload_Tensorflow_model_to_Google_Cloud_Vertex_AI(
39
38
model_path,
40
- tensorflow_version = "2.7", # TODO: Remove the explicit default once the upload_tensorflow_saved_model supports None
39
+ tensorflow_version = None,
41
40
use_gpu = False,
42
41
43
- display_name = "Tensorflow model" ,
42
+ display_name = None ,
44
43
description = None,
45
44
46
45
# Uncomment when anyone requests these:
@@ -51,7 +50,7 @@ implementation:
51
50
# explanation_parameters: "google.cloud.aiplatform_v1.types.explanation.ExplanationParameters" = None,
52
51
53
52
project = None,
54
- location = "us-central1" ,
53
+ location = None ,
55
54
labels = None,
56
55
# encryption_spec_key_name: str = None,
57
56
staging_bucket = None,
0 commit comments