jim/openai-cookbook
1
0
Fork 0
mirror of https://github.com/james-m-jordan/openai-cookbook.git synced 2025-05-09 19:32:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix wget download updated parameters (#538)

Browse Source
This commit is contained in:
Michael Yuan 2023-06-26 04:45:57 -04:00 committed by GitHub
parent de3bd58434
commit 6798e1ca3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/vector_databases/redis/nbutils.py
View File

@ -23,7 +23,7 @@ def download_wikipedia_data(
else: else:
print("File not found, downloading now...") print("File not found, downloading now...")
# Download the data # Download the data
wget.download(data_url, out=download_path, bar=True) wget.download(data_url, out=download_path)
# Unzip the data # Unzip the data
with zipfile.ZipFile(zip_file_path, 'r') as zip_ref: with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
@ -43,4 +43,4 @@ def read_wikipedia_data(data_path: str = '../../data/', file_name: str = "vector
data['content_vector'] = data.content_vector.apply(literal_eval) data['content_vector'] = data.content_vector.apply(literal_eval)
# Set vector_id to be a string # Set vector_id to be a string
data['vector_id'] = data['vector_id'].apply(str) data['vector_id'] = data['vector_id'].apply(str)
return data return data