Enable Grain smoke test

iindyk · copybara-github · commit b04a3a24e866 · 2025-09-03T07:58:28.000-07:00
Ported from #177 PiperOrigin-RevId: 802554031
diff --git 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
 }
 

Original file line number	Diff line number	Diff line change
`@@ -91,8 +91,7 @@ function main() {`
`91`	`91`	`$PYTHON_BIN -c 'import array_record'`
`92`	`92`	`$PYTHON_BIN -c 'from array_record.python import array_record_data_source'`
`93`	`93`	`$PYTHON_BIN -m pip install jax tensorflow>=2.20.0 grain`
`94`		`- # Re-enable the grain import test once the new version is released.`
`95`		`- # $PYTHON_BIN oss/test_import_grain.py`
	`94`	`+ $PYTHON_BIN oss/test_import_grain.py`
`96`	`95`	`$PYTHON_BIN oss/test_import_tensorflow.py`
`97`	`96`	`}`
`98`	`97`