提交 750f9606 编写于 作者: Jayyoung_'s avatar Jayyoung_

fix style bug

上级 4073aebf
...@@ -121,7 +121,7 @@ def try_import_pyarrow(): ...@@ -121,7 +121,7 @@ def try_import_pyarrow():
def try_import_pickle(): def try_import_pickle():
try: try:
import pickle5 as pickle import pickle5 as pickle
assert sys.version_info > (3, 8) assert sys.version_info >= (3, 8)
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
import pickle import pickle
assert sys.version_info < (3, 8) assert sys.version_info < (3, 8)
......
...@@ -63,4 +63,4 @@ class SerializedZoo: ...@@ -63,4 +63,4 @@ class SerializedZoo:
if self.type == 'mp_pyarrow': if self.type == 'mp_pyarrow':
return MultiprocessingPyarrow( return MultiprocessingPyarrow(
self.var, self.shape, self.cpu_count, mode='load', byte_length=self.byte_length self.var, self.shape, self.cpu_count, mode='load', byte_length=self.byte_length
).unpack() ).unpack()
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册