89cc3b59e4c83baf94b96e4b1f8752779009595f,tensorflow_hub/keras_layer_test.py,KerasLayerTest,test_load_with_defaults,#KerasLayerTest#Any#,798

Before Change

      )
  def test_load_with_defaults(self, module_name):
    inputs, expected_outputs = 10., 11.  // Test modules perform increment op.
    path = test_utils.get_test_data_path(module_name)
    layer = hub.KerasLayer(path)
    output = layer(inputs)
    self.assertEqual(output, expected_outputs)

After Change

  @parameterized.parameters(("TF1HubModule"), ("TF2SavedModel_SavedRaw"))
  def test_load_with_defaults(self, model_format):
    export_dir = os.path.join(self.get_temp_dir(), "plus_one_" + model_format)
    _dispatch_model_format(model_format, _save_plus_one_saved_model_v2,
                           _save_plus_one_hub_module_v1, export_dir)
    inputs, expected_outputs = 10., 11.  // Test modules perform increment op.
    layer = hub.KerasLayer(export_dir)
    output = layer(inputs)
    self.assertEqual(output, expected_outputs)

Instances