From 3c7dbef54d074e283ca3103afef3e4a37c2e4f3c Mon Sep 17 00:00:00 2001 From: Fraser Date: Thu, 30 Mar 2023 10:09:43 -0400 Subject: [PATCH] swap to using 600 as dataset --- api/{dataset_dict.pkl => dataset_dict_30.pkl} | 0 api/get_blocks.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename api/{dataset_dict.pkl => dataset_dict_30.pkl} (100%) diff --git a/api/dataset_dict.pkl b/api/dataset_dict_30.pkl similarity index 100% rename from api/dataset_dict.pkl rename to api/dataset_dict_30.pkl diff --git a/api/get_blocks.py b/api/get_blocks.py index 54a78bb..f1c939c 100644 --- a/api/get_blocks.py +++ b/api/get_blocks.py @@ -14,7 +14,7 @@ COMPLETIONS_MODEL = "gpt-3.5-turbo" import pathlib project_path = pathlib.Path(__file__).parent -PATH_TO_DATASET_DICT = project_path / "dataset_dict.pkl" +PATH_TO_DATASET_DICT = project_path / "dataset_dict_500.pkl" with open(PATH_TO_DATASET_DICT, 'rb') as f: data = pickle.load(f)