fix joblib.load from stream

joblib · Mar 23, 2021 · 6a9edf4 · 6a9edf4
1 parent 457d2c8
commit 6a9edf4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/joblib/numpy_pickle_utils.py b/joblib/numpy_pickle_utils.py
@@ -69,8 +69,9 @@ def _detect_compressor(fileobj):
         first_bytes = fileobj.peek(max_prefix_len)
     else:
         # Fallback to seek if the fileobject is not peekable.
+        prev_pos = fileobj.tell()
         first_bytes = fileobj.read(max_prefix_len)
-        fileobj.seek(0)
+        fileobj.seek(prev_pos)
 
     if first_bytes.startswith(_ZFILE_PREFIX):
         return "compat"