diff --git a/cij/cli/extract.py b/cij/cli/extract.py index 09cf708..3be9cdf 100644 --- a/cij/cli/extract.py +++ b/cij/cli/extract.py @@ -5,8 +5,9 @@ def load_data(var): import pandas + from glob import glob - df = pandas.read_table(glob.glob(f"{var}_tp_*")[0], sep="\s+", index_col=0) + df = pandas.read_table(glob(f"{var}_tp_*")[0], sep="\s+", index_col=0) df.columns = [float(colname) for colname in df.columns] df.index = [float(idx) for idx in df.index] return df