From 4c2ddc745547db4bd340b88567d0e6c2f4033b6e Mon Sep 17 00:00:00 2001 From: Nathan TeBlunthuis Date: Sat, 11 Jan 2025 21:50:07 -0800 Subject: [PATCH] bugfix --- similarities/weekly_cosine_similarities.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/similarities/weekly_cosine_similarities.py b/similarities/weekly_cosine_similarities.py index 0121935..f869fbf 100755 --- a/similarities/weekly_cosine_similarities.py +++ b/similarities/weekly_cosine_similarities.py @@ -98,11 +98,11 @@ def cosine_similarities_weekly(tfidf_path, outfile, term_colname, included_subre q = f"SELECT DISTINCT CAST(CAST(week AS DATE) AS STRING) AS week FROM read_parquet('{tfidf_path}/*/*.parquet')" if min_date is not None and max_date is not None: - q = q + f" WHERE CAST(week AS DATE) >= CAST({min_date} AS DATE) AND CAST(week AS DATE) <= CAST({max_date} AS DATE)" + q = q + f" WHERE CAST(week AS DATE) >= CAST('{min_date}' AS DATE) AND CAST(week AS DATE) <= CAST('{max_date}' AS DATE)" elif max_date is not None: - q = q + f" WHERE CAST(week AS DATE) <= CAST({max_date} AS DATE)" + q = q + f" WHERE CAST(week AS DATE) <= CAST('{max_date}' AS DATE)" elif min_date is not None: - q = q + f" WHERE CAST(week AS DATE) >= CAST({min_date} AS DATE)" + q = q + f" WHERE CAST(week AS DATE) >= CAST('{min_date}' AS DATE)" weeks = conn.execute(q).df() weeks = weeks.week.values