From dc221f3bf5a1e411f98e68218ae88e9009f51923 Mon Sep 17 00:00:00 2001
From: Philippe Moussalli <philippe.moussalli95@gmail.com>
Date: Tue, 9 Jan 2024 15:50:11 +0100
Subject: [PATCH] implement small fixes (#60)

---
 src/evaluation.ipynb  | 2 +-
 src/pipeline.ipynb    | 2 +-
 src/pipeline_eval.py  | 2 +-
 src/pipeline_index.py | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/evaluation.ipynb b/src/evaluation.ipynb
index 0f1ec3b..71e4158 100644
--- a/src/evaluation.ipynb
+++ b/src/evaluation.ipynb
@@ -235,7 +235,7 @@
     "import weaviate\n",
     "\n",
     "try:\n",
-    "    local_weaviate_client = weaviate.Client(\"http://localhost:8080\")\n",
+    "    local_weaviate_client = weaviate.Client(\"http://localhost:8081\")\n",
     "    logging.info(\"Connected to Weaviate instance\")\n",
     "except weaviate.WeaviateStartUpError:\n",
     "    logging.error(\"Cannot connect to weaviate instance, is it running?\")"
diff --git a/src/pipeline.ipynb b/src/pipeline.ipynb
index a7f4552..87cd446 100644
--- a/src/pipeline.ipynb
+++ b/src/pipeline.ipynb
@@ -202,7 +202,7 @@
     "        accelerator_number=number_of_accelerators,\n",
     "        accelerator_name=accelerator_name,\n",
     "    ),\n",
-    "    cluster_type=\"local\"\n",
+    "    cluster_type=\"local\" if number_of_accelerators is not None else \"default\",\n",
     ")\n",
     "\n",
     "embeddings.write(\n",
diff --git a/src/pipeline_eval.py b/src/pipeline_eval.py
index 23e222b..cf23ec1 100644
--- a/src/pipeline_eval.py
+++ b/src/pipeline_eval.py
@@ -54,7 +54,7 @@ def create_pipeline(
             accelerator_number=number_of_accelerators,
             accelerator_name=accelerator_name,
         ),
-        cluster_type="local",
+        cluster_type="local" if number_of_accelerators is not None else "default",
     )
 
     retrieve_chunks = embed_text_op.apply(
diff --git a/src/pipeline_index.py b/src/pipeline_index.py
index d5f8748..de76060 100644
--- a/src/pipeline_index.py
+++ b/src/pipeline_index.py
@@ -56,7 +56,7 @@ def create_pipeline(
             accelerator_number=number_of_accelerators,
             accelerator_name=accelerator_name,
         ),
-        cluster_type="local",
+        cluster_type="local" if number_of_accelerators is not None else "default",
     )
 
     embeddings.write(