suse-model-store/models/linear/linear.py

from pandas import DataFrame

from models.models.linear.operations import operate


class Linear:
    def __init__(self):
        self.weight = 0.2
        self.bias = 5

    def predict(self, x: DataFrame) -> DataFrame:
        return operate(x, self.weight, self.bias)