From 75a45a0e6d8f48d3e2a35e8db92c9b7820846011 Mon Sep 17 00:00:00 2001 From: jordan Date: Thu, 21 Sep 2023 14:13:43 +0000 Subject: [PATCH] fix dependencies --- README.md | 6 ++++++ src/vertex/helpers.py | 9 ++++++--- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index f0d8ae2..cb68a76 100644 --- a/README.md +++ b/README.md @@ -119,6 +119,12 @@ To run the notebooks successfully, follow the steps below. #### Dependencies +**(TODO) General:** +``` +python-decouple +``` + +**Tensorflow:** ``` tensorflow==2.11.0 tensorflow-cloud==0.1.16 diff --git a/src/vertex/helpers.py b/src/vertex/helpers.py index 17393ae..66d8e2d 100644 --- a/src/vertex/helpers.py +++ b/src/vertex/helpers.py @@ -71,9 +71,12 @@ def _build_index_config(embedding_gcs_uri: str, dimensions: int, index_type: str ) else: - except Exception as e: - logger.error(f'Failed to create index. select either "ann" or "bf" for "index_type"') - raise e + logger.error(f'Failed to create index. select either "ann" or "bf" for "index_type"') + + # TODO - add try / except logic: + # except Exception as e: + # logger.error(f'Failed to create index. select either "ann" or "bf" for "index_type"') + # raise e return metadata