From efaad495b677a67b8825a028dc155d9cf07c0be8 Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Tue, 28 Nov 2023 13:36:30 -0500 Subject: [PATCH] Fix lint --- .../IMDB_extended/training_data_collection_telemetry.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/workloads/IMDB_extended/training_data_collection_telemetry.py b/workloads/IMDB_extended/training_data_collection_telemetry.py index 666d0e42..4d9f1943 100644 --- a/workloads/IMDB_extended/training_data_collection_telemetry.py +++ b/workloads/IMDB_extended/training_data_collection_telemetry.py @@ -1,6 +1,5 @@ import copy import json -import os.path from tqdm import tqdm import argparse @@ -243,6 +242,8 @@ def simulate_query_on_larger_scale( ) if target_path is not None: - with open(target_path + f"_epoch_{target_scale}.json", "w") as f: + with open( + target_path + f"_epoch_{target_scale}.json", "w", encoding="UTF-8" + ) as f: json.dump(parsed_queries, f, default=dumper) return parsed_queries