docstr: Tests that the model can be compiled and run a forward pass.
docstr_tok: ["Tests","that","the","model","can","be","compiled","and","run","a","forward","pass","."]
code: def test_warm_up(self):
      """Tests that the model can be compiled and run a forward pass."""
      self.model.warm_up()
  
file_path: scenic/scenic/projects/owl_vit/notebooks/tests/inference_test.py
imports: from unittest import mock
  from absl.testing import absltest
  import chex
  import jax
  import jax.numpy
  import numpy
  from scenic.projects.owl_vit import models
  from scenic.projects.owl_vit.configs import clip_b32
  from scenic.projects.owl_vit.notebooks import inference
label_desc: Use image node_id 2 for calling the InferenceTest obj's underlying member method code with example usage: obj.test_warm_up() without return types