Generalise file access

alisw · Nov 27, 2023 · 7034555 · 7034555
1 parent f29b4a8
commit 7034555
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/machine_learning_hep/utilities.py b/machine_learning_hep/utilities.py
@@ -73,6 +73,8 @@ def openfile(filename, attr):
         return gzip.open(filename, attr)
     if filename.lower().endswith('.lz4'):
         return lz4.frame.open(filename, attr)
+    if filename.lower().endswith('.pkl'):
+        return open(filename, attr, encoding='utf-8')
     return open(filename, attr)
 
 def mask_df(df_to_mask, mask_config):