From a9b296dd73889a853257d6478ec0f5aae41534c8 Mon Sep 17 00:00:00 2001 From: Nathan TeBlunthuis Date: Sat, 28 Dec 2024 20:18:53 -0800 Subject: [PATCH] bugfix --- similarities/similarities_helper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/similarities/similarities_helper.py b/similarities/similarities_helper.py index 47c2664..0050c1d 100644 --- a/similarities/similarities_helper.py +++ b/similarities/similarities_helper.py @@ -16,6 +16,7 @@ import pathlib from datetime import datetime from pathlib import Path import pickle +import pytz class tf_weight(Enum): MaxTF = 1 @@ -186,7 +187,7 @@ def write_weekly_similarities(path, sims, week, names): # reformat as a pairwise list sims = sims.melt(id_vars=['_subreddit','week'],value_vars=names.subreddit.values) - sims.to_parquet(p / week) + sims.to_parquet(p / week.item().replace(tzinfo=pytz.UTC).isoformat('T')) def column_overlaps(mat): non_zeros = (mat != 0).astype('double')