2909332cb7d37f80b6f7096043a3784e5c9f4aa9,hyperopt/gdist.py,Union,perform,#Union#Any#Any#Any#,47

Before Change

    def perform(self, node, inputs, outstorage):
        ans = set(inputs[0])
        for ii in inputs[1:]:
            ans.update(ii)
        ans = numpy.array(sorted(ans))
        outstorage[0][0] = node.outputs[0].type.filter(ans, strict=False)

union = Union()

After Change

    def perform(self, node, inputs, outstorage):
        for ii in inputs:
            if "int" not in str(ii.dtype):
                raise TypeError("non-int arg to SetOp", ii)
        ans = set(inputs[0])
        if self.operation == "union":
            ans.update(*inputs[1:])
        elif self.operation == "difference":
            ans.difference_update(*inputs[1:])
        else:
            raise NotImplementedError(self.operation)
        npy_ans = numpy.array(sorted(ans), dtype=node.outputs[0].dtype)
        outstorage[0][0] = node.outputs[0].type.filter(npy_ans, strict=True)

union = SetOp("union")

Instances