diff --git a/paper/src/context_threads.py b/paper/src/context_threads.py index 0d6fd53..93a635d 100644 --- a/paper/src/context_threads.py +++ b/paper/src/context_threads.py @@ -210,7 +210,7 @@ def range_of_dates(start_date, extra_days): # write contests to ndjson with open( os.path.join( - f"tweet_threads_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() }_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() + datetime.timedelta(days=args.extra_days-1)}.ndjson", + f"tweet_threads_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date()}_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() + datetime.timedelta(days=args.extra_days-1)}.ndjson", ), "w", ) as f: diff --git a/src/conspiracies/docprocessing/relationextraction/gptprompting/prompt_apis.py b/src/conspiracies/docprocessing/relationextraction/gptprompting/prompt_apis.py index 8f0e7fb..6144353 100644 --- a/src/conspiracies/docprocessing/relationextraction/gptprompting/prompt_apis.py +++ b/src/conspiracies/docprocessing/relationextraction/gptprompting/prompt_apis.py @@ -4,7 +4,9 @@ from spacy.tokens import Doc -from conspiracies.docprocessing.relationextraction.gptprompting.prompt_templates import PromptTemplate +from conspiracies.docprocessing.relationextraction.gptprompting.prompt_templates import ( + PromptTemplate, +) from conspiracies.registry import registry diff --git a/src/conspiracies/preprocessing/create_context_threads.py b/src/conspiracies/preprocessing/create_context_threads.py index 2301d9e..e7a0acf 100644 --- a/src/conspiracies/preprocessing/create_context_threads.py +++ b/src/conspiracies/preprocessing/create_context_threads.py @@ -135,7 +135,7 @@ def range_of_dates(start_date, extra_days): # write contests to ndjson; the ndjson of contexts has the format: List[Dict[str, Any]] with open( os.path.join( - f"tweet_threads_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() }_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() + datetime.timedelta(days=args.extra_days-1)}.ndjson", + f"tweet_threads_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date()}_{datetime.datetime.strptime(args.start_date, '%Y-%m-%d').date() + datetime.timedelta(days=args.extra_days-1)}.ndjson", ), "w", ) as f: