Squeeze imdb test more to fit travis

PiperOrigin-RevId: 337800342

examples/imdb.py

@@ -111,7 +111,7 @@ def main(*args, use_dummy_data: bool = False, **kwargs) -> None:
 def _get_dummy_data(mask_constant: float
                     ) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]:
   """Return dummy data for when downloading embeddings is not feasible."""
-  n_train, n_test = 10, 10
+  n_train, n_test = 5, 5
 
   def get_x(shape, key):
     key_x, key_mask = random.split(key)