WIP: fix

Author: tellet-q

.github/workflows/continuous-benchmark-transfer.yaml

@@ -5,7 +5,7 @@ on:
     inputs:
       dataset_name:
         description: 'Dataset name'
-        default: 'dbpedia-openai-1M-1536-angular'
+        default: 'dbpedia-openai-100K-1536-angular'
         type: string
       qdrant_versions:
         description: 'Comma-separated versions (ghcr/dev, docker/master, docker/v1.13.0)'
@@ -38,7 +38,7 @@ env:
   SERVER_TYPE: ${{ inputs.server_type || 'cpx42' }}
   CLIENT_TYPE: ${{ inputs.client_type || 'cpx42' }}
   REGION: ${{ inputs.region || 'fsn1' }}
-  DATASET_NAME: ${{ inputs.dataset_name || 'dbpedia-openai-1M-1536-angular' }}
+  DATASET_NAME: ${{ inputs.dataset_name || 'dbpedia-openai-100K-1536-angular' }}
   QDRANT_VERSIONS: ${{ inputs.qdrant_versions || 'ghcr/dev,docker/master' }}
 
 jobs:

ansible/playbooks/roles/run-transfer-speed/files/shard_transfer.py

@@ -104,7 +104,7 @@ def upload_vectors(self, vectors: np.ndarray):
             optimizer_config=models.OptimizersConfigDiff(indexing_threshold=1),
         )
 
-    def wait_for_green(self, timeout: int = 600):
+    def wait_for_green(self, timeout: int = 1800):
         print("Waiting for green status...", end="", flush=True)
         start = time.time()
         while time.time() - start < timeout: