Skip to content

Commit

Permalink
implement small fixes (#60)
Browse files Browse the repository at this point in the history
  • Loading branch information
PhilippeMoussalli authored Jan 9, 2024
1 parent 5ac5ebd commit dc221f3
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion src/evaluation.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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?\")"
Expand Down
2 changes: 1 addition & 1 deletion src/pipeline.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
2 changes: 1 addition & 1 deletion src/pipeline_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down
2 changes: 1 addition & 1 deletion src/pipeline_index.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit dc221f3

Please sign in to comment.