diff --git a/python/parfield/dorie/parfield/parameter_file.py b/python/parfield/dorie/parfield/parameter_file.py index 37d81d48d19ab2c3a3a180a8080b60ba0d997b4b..11e4d732378c348930bd8f32c88e4b8d140e23c8 100644 --- a/python/parfield/dorie/parfield/parameter_file.py +++ b/python/parfield/dorie/parfield/parameter_file.py @@ -2,6 +2,11 @@ import os import h5py import numpy as np +try: # Python 2.x compatibility + input = raw_input +except NameError: + pass + def read_parameter_file(in_path): data_dict = {} with h5py.File(in_path, 'r') as f: @@ -32,7 +37,7 @@ def write_parameter_file(group,data_dict,out_path,attrs=None,overwrite=False): """ if os.path.isfile(out_path) and not overwrite: - ow = raw_input("The output file {} already exists. Overwrite? [y/N] ".format(out_path)) + ow = input("The output file {} already exists. Overwrite? [y/N] ".format(out_path)) if not ow.lower() in ["y","yes"]: raise RuntimeError("Aborting")