fix: missing install

.cloud-build/execute_changed_notebooks_helper.py

@@ -32,7 +32,8 @@ import execute_notebook_helper
 import execute_notebook_remote
 import nbformat
 from google.cloud.devtools.cloudbuild_v1.types import BuildOperationMetadata
-from ratemate import RateLimit
+from pyrate_limiter import (Duration, RequestRate,
+                            Limiter, FileLockSQLiteBucket)
 from tabulate import tabulate
 from utils import NotebookProcessors, util
 
@@ -156,9 +157,10 @@ def _create_tag(filepath: str) -> str:
     return tag
 
 
-rate_limit = RateLimit(max_count=10, per=60, greedy=True)
-
+rate_limit = RequestRate(10, Duration.HOUR)
+limiter = Limiter(rate_limit, bucket_class=FileLockSQLiteBucket)
 
+@limiter.ratelimit('testing', delay=True, max_delay=360)
 def process_and_execute_notebook(
     container_uri: str,
     staging_bucket: str,
@@ -172,7 +174,6 @@ def process_and_execute_notebook(
     notebook: str,
     should_get_tail_logs: bool = False,
 ) -> NotebookExecutionResult:
-    rate_limit.wait()  # wait before creating the task
 
     print(f"Running notebook: {notebook}")
 

.cloud-build/requirements.txt

@@ -9,6 +9,7 @@ tabulate
 google-cloud-aiplatform
 google-cloud-storage
 google-cloud-build
-ratemate
 GitPython
-tqdm
+tqdm
+pyrate-limiter
+filelock