af461df627ef660d1a71b6981dedb4e4b504ba9a,txtgen/modules/connectors/connectors.py,StochasticConnector,_build,#StochasticConnector#Any#,256

Before Change

            A Tensor or a (nested) tuple of Tensors of the same structure of
            the decoder state.
        
        modules = ["txtgen.custom", "txtgen.core.distributions"]
        sampler = get_function(self.hparams.distribution, modules)

        if sampler is distributions.sample_gaussian:
            if not isinstance(inputs, tuple):

After Change

        try:
            nest.assert_same_structure(inputs, self._decoder_state_size)
        except (ValueError, TypeError):
            flat_input = nest.flatten(inputs)
            output = nest.pack_sequence_as(
                self._decoder_state_size, flat_input)

        self._add_internal_trainable_variables()
        self._built = True

Instances