code: def test_first_time_chief_generate_nonadaptive(self, import_fn):
      import_fn.side_effect = _fake_import_fn
      generator = replay_generator.ReplayGenerator(
          phoenix_spec=_create_spec("nonadaptive"),
          metadata=ml_metadata_db.MLMetaData(
              _create_spec("nonadaptive"),
              study_owner="test",
              study_name="test",
          ),
      )
      call_args = generator.first_time_chief_generate(
          input_layer_fn="input_layer",
          trial_mode="mode",
          logits_dimension=5,
          hparams=None,
          run_config=tf.estimator.RunConfig(model_dir="mydir/5"),
          is_training=True,
          trials=[],
      )
      self.assertEqual(
          call_args,
          {
              "phoenix_spec": _create_spec("nonadaptive"),
              "is_training": True,
              "logits_dimension": 5,
              "previous_model_dirs": [
                  "mydir/1",
                  "mydir/2",
                  "mydir/3",
                  "mydir/4",
              ],
              "force_freeze": True,
              "allow_auxiliary_head": False,
              "caller_generator": "replay_generator",
              "my_model_dir": "mydir/5",
          },
      )
  
file_path: model_search/model_search/generators/replay_generator_test.py
imports: from absl.testing import absltest
  import mock
  from model_search.generators import replay_generator
  from model_search.metadata import ml_metadata_db
  from model_search.proto import ensembling_spec_pb2
  from model_search.proto import phoenix_spec_pb2
  import tensorflow.compat.v2
label_desc: Use image node_id 2 for calling the ReplayGeneratorTest obj's underlying member method code with example usage: obj.test_first_time_chief_generate_nonadaptive(import_fn) without return types