75284b46aca2d78bcc1b80739b2ed283b8ca457d,tensorforce/models/distribution_model.py,DistributionModel,get_components,#DistributionModel#,274

Before Change

        return model_summaries + network_summaries + distribution_summaries

    def get_components(self):
        model_components = super(DistributionModel, self).get_components()
        return model_components + [self.network] + list(self.distributions.values())

After Change

        return model_summaries + network_summaries + distribution_summaries

    def get_components(self):
        result = dict(super(DistributionModel, self).get_components())
        result[DistributionModel.COMPONENT_NETWORK] = self.network
        for action, distribution in self.distributions.items():
            result["%s_%s" % (DistributionModel.COMPONENT_DISTRIBUTION, action)] = distribution
        if len(self.distributions) == 1:
            result[DistributionModel.COMPONENT_DISTRIBUTION] = next(iter(self.distributions.values()))
        return result

Instances