1b3de6b301f962db0271fdad1bbb5036880b7fbc,pycox/preprocessing/label_transforms.py,LabTransPCHazard,__init__,#LabTransPCHazard#,128

Before Change

class LabTransPCHazard:
    def __init__(self, cuts, min_=0., dtype=None):
        self._predefined_cuts = False
        if type(cuts) is int:
            cuts = ("equidistant", cuts + 1)
        elif hasattr(cuts, "__iter__"):
            if (type(cuts[0]) is not str):
                self.idu = IdxDiscUnknownC(cuts)
                assert dtype is None, "Need dtype to be None for spesified cuts"
                self.dtype = type(cuts[0])
                self._dtype = self.dtype
                self._predefined_cuts = True
            elif (len(cuts) == 2) and (type(cuts[0]) is str):
                cuts = (cuts[0], cuts[1] + 1)
        self._cuts = cuts
        self.min_ = min_
        self._dtype = dtype

After Change

            self.cuts = cuts
            self.idu = IdxDiscUnknownC(self.cuts)
            assert dtype is None, "Need `dtype` to be `None` for specified cuts"
            self._dtype = type(self.cuts[0])
            self._dtype_init = self._dtype
            self._predefined_cuts = True
        else:
            self._cuts += 1

Instances