From b04a3a24e866d90865a6d7b5686951e516a960ac Mon Sep 17 00:00:00 2001 From: Ihor Indyk Date: Wed, 3 Sep 2025 07:57:57 -0700 Subject: [PATCH] Enable Grain smoke test Ported from https://github.com/google/array_record/pull/177 PiperOrigin-RevId: 802554031 --- oss/build_whl.sh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/oss/build_whl.sh b/oss/build_whl.sh index 85e1cad..d6bf018 100755 --- a/oss/build_whl.sh +++ b/oss/build_whl.sh @@ -91,8 +91,7 @@ function main() { $PYTHON_BIN -c 'import array_record' $PYTHON_BIN -c 'from array_record.python import array_record_data_source' $PYTHON_BIN -m pip install jax tensorflow>=2.20.0 grain - # Re-enable the grain import test once the new version is released. - # $PYTHON_BIN oss/test_import_grain.py + $PYTHON_BIN oss/test_import_grain.py $PYTHON_BIN oss/test_import_tensorflow.py }