From a755ecc4801f47379b7a389bc4eaae96c0725527 Mon Sep 17 00:00:00 2001 From: Thomas Lemoine Date: Wed, 21 Jun 2023 00:36:14 -0400 Subject: [PATCH] removed PATH_TO_DATASET_PKL, stored in dict now --- src/dataset/create_dataset.py | 2 +- src/main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dataset/create_dataset.py b/src/dataset/create_dataset.py index 516e4cb..bdf5182 100644 --- a/src/dataset/create_dataset.py +++ b/src/dataset/create_dataset.py @@ -21,7 +21,7 @@ except ImportError: openai.api_key = os.environ.get('OPENAI_API_KEY') -from .settings import PATH_TO_DATASET_PKL, PATH_TO_DATASET_DICT_PKL, EMBEDDING_MODEL, LEN_EMBEDDINGS +from .settings import PATH_TO_DATASET_DICT_PKL, EMBEDDING_MODEL, LEN_EMBEDDINGS from .text_splitter import TokenSplitter, split_into_sentences diff --git a/src/main.py b/src/main.py index 333795a..029df5a 100644 --- a/src/main.py +++ b/src/main.py @@ -27,7 +27,7 @@ if str(src_path) not in sys.path: from dataset import create_dataset #from assistant import semantic_search -from settings import EMBEDDING_MODEL, PATH_TO_DATASET_DICT_PKL, PATH_TO_DATASET_PKL +from settings import EMBEDDING_MODEL, PATH_TO_DATASET_DICT_PKL import numpy as np import matplotlib.pyplot as plt