Merge pull request inspirehep#2374 from jacquerie/integrate-pr-2372

workflows: fix arxiv_plot_extract
kaplun · May 23, 2017 · 1427bc0 · 1427bc0
2 parents 0208ed2 + e166d03
commit 1427bc0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inspirehep/modules/workflows/tasks/arxiv.py b/inspirehep/modules/workflows/tasks/arxiv.py
@@ -31,7 +31,6 @@
 import backoff
 from flask import current_app
 from lxml.etree import XMLSyntaxError
-from six import BytesIO
 from wand.exceptions import DelegateError
 from werkzeug import secure_filename
 
@@ -131,7 +130,8 @@ def arxiv_plot_extract(obj, eng):
             return
 
         for idx, plot in enumerate(plots):
-            obj.files[plot.get('name')] = BytesIO(open(plot.get('url')))
+            with open(plot.get('url')) as plot_file:
+                obj.files[plot.get('name')] = plot_file
             obj.files[plot.get('name')]['description'] = u'{0:05d} {1}'.format(
                 idx, ''.join(plot.get('captions', []))
             )