groceryheist/cdsc_reddit
1
0
Fork 0
Code

bugfix

Browse Source
This commit is contained in:
Nathan TeBlunthuis 2025-01-12 00:54:34 -08:00
parent 8f0ce2dba7
commit 1fa2f6c4d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
similarities/weekly_cosine_similarities.py
View File

@ -46,7 +46,7 @@ def _week_similarities(week, simfunc, tfidf_path, term_colname, included_subredd
tfidf_colname='tf_idf' tfidf_colname='tf_idf'
if term_ids is not None: if term_ids is not None:
entries = duckdb.sql(f"SELECT A.{tfidf_colname}, B.{term_id} AS {term_id_new}, A.subreddit_id_new FROM entries AS A JOIN term_ids AS B ON A.{term_id_new} == B.old_+{term_id}").df() entries = duckdb.sql(f"SELECT A.{tfidf_colname}, B.{term_id} AS {term_id_new}, A.subreddit_id_new FROM entries AS A JOIN read_parquet('{term_ids}') AS B ON A.{term_id_new} == B.old_+{term_id}").df()