def get_point()

in captum/_utils/models/linear_model/train.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

    def get_point(datapoint):
        if len(datapoint) == 2:
            x, y = datapoint
            w = None
        else:
            x, y, w = datapoint

        if device is not None:
            x = x.to(device)
            y = y.to(device)
            if w is not None:
                w = w.to(device)

        return x, y, w