Fix iree-samples buildkite pipeline. (#8142)
Might drop or rewrite the notebooks using TensorFlow soon anyways (https://github.com/google/iree/issues/8086)
diff --git a/colab/test_notebooks.py b/colab/test_notebooks.py
index 2de3de6..af545b8 100644
--- a/colab/test_notebooks.py
+++ b/colab/test_notebooks.py
@@ -14,7 +14,13 @@
NOTEBOOKS_TO_SKIP = []
-NOTEBOOKS_EXPECTED_TO_FAIL = []
+NOTEBOOKS_EXPECTED_TO_FAIL = [
+ # Logs: https://buildkite.com/iree/iree-samples/builds/212#2d108f5a-b7d1-46de-93d3-85ce60250514
+ # error: 'util.global.store' op global type mismatch;
+ # global __iree_flow___sm_node14__optimizer.iter is 'tensor<i64>' but
+ # store is 'tensor<i32>'
+ "mnist_training.ipynb",
+]
class ColabNotebookTests(absltest.TestCase):
