Python equals Examples

Programming Language: Python

Namespace/Package Name: gluonts.torch.component

Method/Function: equals

Examples at hotexamples.com: 2

Python equals - 2 examples found. These are the top rated real world Python examples of gluonts.torch.component.equals extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

    def __eq__(self, that):
        if type(self) != type(that):
            return False

        if not equals(self.input_transform, that.input_transform):
            return False

        return equals(
            self.prediction_net.state_dict(),
            that.prediction_net.state_dict(),
        )

Example #2

Show file

    def __eq__(self, that):
        if type(self) != type(that):
            return False

        # TODO: also consider equality of the pipelines
        # if not equals(self.input_transform, that.input_transform):
        #    return False

        return equals(
            self.prediction_net.state_dict(),
            that.prediction_net.state_dict(),
        )