code: def test_feature_pyramid_encoder_no_text_token(
      self, input_mask, window_size, window_block_indexes
  ):
      rng = random.PRNGKey(0)
      x = np.ones((2, 8, 8), dtype=np.float32)
      output, params = video_text_fusion.FeaturePyramidEncoder(
          num_layers=4,
          mlp_dim=16,
          num_heads=2,
          window_size=window_size,
          window_block_indexes=window_block_indexes,
          feature_pyramid_config=ml_collections.ConfigDict(
              {
                  "num_features_level0": 8,
                  "feature_pyramid_levels": [2, 3],
                  "feature_pyramid_downsample_stride": 2,
              }
          ),
      ).init_with_output(rng, x, input_mask, train=False)
      self.assertTupleEqual(output.shape, (2, 12, 8))
      self.assertSetEqual(
          set(params["params"].keys()),
          {
              "posembed_input",
              "encoderblock_0",
              "encoderblock_1",
              "encoderblock_2",
              "encoderblock_3",
              "output_conv_0",
              "output_conv_1",
              "output_ln_0",
              "output_ln_1",
          },
      )
  
file_path: scenic/scenic/projects/unloc/video_text_fusion_test.py
imports: from absl.testing import parameterized
  from jax import random
  import ml_collections
  import numpy
  from scenic.projects.unloc import video_text_fusion
  import tensorflow
label_desc: Use image node_id 5 for calling the LayersTest obj's underlying member method code with example usage: obj.test_feature_pyramid_encoder_no_text_token(input_mask, window_size, window_block_indexes) without return types