uber / petastorm

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.
Apache License 2.0
1.78k stars 285 forks source link

Fix mypy test: removing np.bool from tf_utils numpy-to-tf dtype conversion #658

Closed selitvin closed 3 years ago

selitvin commented 3 years ago

A proper numpy dtype to use is np.bool_ and not np.bool. np.bool_ is already present in the numpy/tf mapping table. This fixes mypy static check that would fail if we upgrade pyarrow to version 2 or higher.