Commit 5b9c0411 authored by Wouter Kessels's avatar Wouter Kessels

Check extension

parent 43410cda
......@@ -17,6 +17,10 @@ def plots(file, settings_passed):
Settings for the interacting system and visual output
"""
# Check whether file extension is given
if file[-4:] != ".pkl":
file += ".pkl"
with open(file, 'rb') as f:
data = pickle.load(f)
f.close()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment