raannakasturi commited on
Commit
246c816
·
1 Parent(s): 11659c8

Fix citation generation by trimming whitespace and update example ID

Browse files
Files changed (1) hide show
  1. fetch_arxiv_data.py +2 -2
fetch_arxiv_data.py CHANGED
@@ -28,9 +28,9 @@ def generate_citation(id):
28
  doi = fetch_pmc_doi(id)
29
  else:
30
  doi = fetch_arxiv_doi(id)
31
- citation = fetch_citation(doi).replace('\n', ' ').replace('<i>', '').replace('</i>', '')
32
  return citation
33
 
34
  if __name__ == '__main__':
35
- citation = generate_citation('2412.14338')
36
  print(citation)
 
28
  doi = fetch_pmc_doi(id)
29
  else:
30
  doi = fetch_arxiv_doi(id)
31
+ citation = fetch_citation(doi).replace('\n', ' ').replace('<i>', '').replace('</i>', '').strip()
32
  return citation
33
 
34
  if __name__ == '__main__':
35
+ citation = generate_citation('2105.03824')
36
  print(citation)