yulongchen commited on
Commit
4a21dec
·
1 Parent(s): dd88909

Add system

Browse files
Files changed (1) hide show
  1. system/process_time.py +7 -9
system/process_time.py CHANGED
@@ -238,23 +238,21 @@ def extract_and_sort_events(data_dir, pledge_date, pledge_author, claim, suggest
238
  mete_date = doc["date"]
239
  for event in doc.get("output", {}).get("events", []):
240
  parsed_date, original_date = parse_date(event["date"])
241
- parsed_date_str = parsed_date.strftime("%Y-%m-%d")
242
 
243
  if parsed_date:
244
- # if mete_date!= parsed_date:
245
- event_date_and_pub_date = parsed_date_str+f" ({mete_date})"
246
- # else:
247
- # event_date_and_pub_date = original_date
248
-
249
- test_instance = f"Pledge: {pledge} (Speaker: {pledge_author}; Pledge Date: {pledge_date})\nEvent Summary: {event['event']} (Event Date: {original_date})\nIs this event summary useful to track the fulfilment of this pledge"
250
 
251
- # print(test_instance)
252
 
253
  label, score = gpt_eval(test_instance, train_data, instruction, suggestion_meta, ICL_id=ICL_id)
254
 
255
  URL = doc["url"]
256
  events.append({
257
- "date": original_date,
258
  "event date (publication date if different)": event_date_and_pub_date,
259
  "event": event["event"],
260
  "url": URL,
 
238
  mete_date = doc["date"]
239
  for event in doc.get("output", {}).get("events", []):
240
  parsed_date, original_date = parse_date(event["date"])
 
241
 
242
  if parsed_date:
243
+ parsed_date_str = parsed_date.strftime("%Y-%m-%d")
244
+ if parsed_date_str != mete_date:
245
+ event_date_and_pub_date = f"{parsed_date_str} ({mete_date})"
246
+ else:
247
+ event_date_and_pub_date = parsed_date_str
 
248
 
249
+ test_instance = f"Pledge: {pledge} (Speaker: {pledge_author}; Pledge Date: {pledge_date})\nEvent Summary: {event['event']} (Event Date: {original_date})\nIs this event summary useful to track the fulfilment of this pledge"
250
 
251
  label, score = gpt_eval(test_instance, train_data, instruction, suggestion_meta, ICL_id=ICL_id)
252
 
253
  URL = doc["url"]
254
  events.append({
255
+ "date": original_date,
256
  "event date (publication date if different)": event_date_and_pub_date,
257
  "event": event["event"],
258
  "url": URL,