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

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)
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