fix(io): fix summary loading

kmnhan · Apr 10, 2024 · a5dd84a · a5dd84a
1 parent 355d023
commit a5dd84a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/erlab/io/dataloader.py b/src/erlab/io/dataloader.py
@@ -428,11 +428,10 @@ def summarize(
         if usecache:
             try:
                 df = pandas.read_pickle(pkl_path)
+                df = df.head(len(df))
             except FileNotFoundError:
                 pass
 
-        df = df.head(len(df))
-
         if df is None:
             df = self.generate_summary(data_dir, **kwargs)
             df.to_pickle(pkl_path)