update pipe

jswortz · Sep 19, 2023 · 10faf18 · 10faf18
1 parent 9a32e21
commit 10faf18
Show file tree

Hide file tree

Showing 14 changed files with 448 additions and 469 deletions.
diff --git a/05-candidate-generation.ipynb b/05-candidate-generation.ipynb
diff --git a/06-matching-engine.ipynb b/06-matching-engine.ipynb
diff --git a/07-train-pipeline.ipynb b/07-train-pipeline.ipynb
diff --git a/src/train_pipes/create_ann_index.py b/src/train_pipes/create_ann_index.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0'
         # 'google-cloud-storage',
     ],

diff --git a/src/train_pipes/create_ann_index_endpoint_vpc.py b/src/train_pipes/create_ann_index_endpoint_vpc.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0',
     ],
 )

diff --git a/src/train_pipes/create_brute_force_index.py b/src/train_pipes/create_brute_force_index.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0',
         # 'google-cloud-storage',
     ],

diff --git a/src/train_pipes/create_brute_index_endpoint_vpc.py b/src/train_pipes/create_brute_index_endpoint_vpc.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0',
     ],
 )

diff --git a/src/train_pipes/create_tensorboard.py b/src/train_pipes/create_tensorboard.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image='python:3.9',
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'numpy',
         'google-cloud-storage',
     ],

diff --git a/src/train_pipes/deploy_ann_index.py b/src/train_pipes/deploy_ann_index.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0',
     ]
 )

diff --git a/src/train_pipes/deploy_brute_index.py b/src/train_pipes/deploy_brute_index.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-api-core==2.11.0',
     ],
 )

diff --git a/src/train_pipes/generate_candidates.py b/src/train_pipes/generate_candidates.py
@@ -8,8 +8,8 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
-        'tensorflow==2.10.1',
+        'google-cloud-aiplatform==1.26.1',
+        'tensorflow==2.11.0',
         'tensorflow-recommenders==0.7.2',
         'numpy',
         # 'google-cloud-storage',

diff --git a/src/train_pipes/test_model_index_endpoint_v5.py b/src/train_pipes/test_model_index_endpoint_v5.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image="python:3.9",
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         'google-cloud-pipeline-components',
         'google-cloud-storage',
         'tensorflow==2.11.0',

diff --git a/src/train_pipes/train_custom_model.py b/src/train_pipes/train_custom_model.py
@@ -8,7 +8,7 @@
 @kfp.v2.dsl.component(
     base_image='python:3.9',
     packages_to_install=[
-        'google-cloud-aiplatform==1.23.0',
+        'google-cloud-aiplatform==1.26.1',
         # 'tensorflow==2.9.2',
         # 'tensorflow-recommenders==0.7.0',
         'numpy',

diff --git a/src/two_tower_jt/test_instances.py b/src/two_tower_jt/test_instances.py
@@ -36,8 +36,8 @@
     'pl_collaborative_src': 'false',
     'pl_duration_ms_new': 17971314.0,
     'pl_name_src': 'Capoeira',
-    'time_signature_can': '4',
-    'time_signature_pl': ['4', '4', '4', '4', '4'],
+    'track_time_signature_can': '4',
+    'track_time_signature_pl': ['4', '4', '4', '4', '4'],
     'track_acousticness_can': 0.478,
     'track_acousticness_pl': [0.238 , 0.105 , 0.0242, 0.125 , 0.304 ],
     'track_danceability_can': 0.709,
@@ -150,8 +150,8 @@
     'pl_collaborative_src': 'false',
     'pl_duration_ms_new': 17971314.0,
     'pl_name_src': 'Capoeira',
-    'time_signature_can': '4',
-    'time_signature_pl': ['4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4'],
+    'track_time_signature_can': '4',
+    'track_time_signature_pl': ['4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4', '4'],
     'track_acousticness_can': 0.478,
     'track_acousticness_pl': [0.238 , 0.105 , 0.0242, 0.125 , 0.304, 0.238 , 0.105 , 0.0242, 0.125 , 0.304, 0.238 , 0.105 , 0.0242, 0.125 , 0.304 ],
     'track_danceability_can': 0.709,