81092be4e1e900473cd008fec50dc49c8af2eed9,tensorflow_hub/keras_layer.py,KerasLayer,call,#KerasLayer#Any#Any#,166

Before Change

    // Unwrap dicts returned by signatures.
    if self._is_hub_module_v1:
      if "default" in result:
        result = result["default"]
      else:
        raise ValueError("Signature does not has an output named `default`.")

    // TODO(b/142213824): Remove setting shapes when shape inference works.
    result = self._apply_output_shape_if_set(inputs, result)
    return result

  def _get_callable(self):

After Change

    // Unwrap dicts returned by signatures.
    if self._output_key:
      if not isinstance(result, dict):
        raise ValueError("Specifying `output_key` is forbidden if output "
                         "type %s is not a dict." % type(result))
      if self._output_key not in result:
        raise ValueError(
            "KerasLayer output does not contain the output key %s "
            "(available: %s)." % (self._output_key, result.keys()))

Instances