1672b5a9a47b081aa3e61c9f2ca76fae26ed8730,cheml/wrappers/cheml_cml/cheml_wrapper.py,SaveFile,fit,#SaveFile#,663

Before Change

    def fit(self):
        // step1: check inputs
        df, df_info = self.input_check("df", req=True, py_type=pd.DataFrame)

        // step2: assign inputs to parameters if necessary (param = @token)
        self.paramFROMinput()

        // step3: check the dimension of input data frame
        // step4: import module and make APIs
        try:
            from cheml.initialization import SaveFile
            model = SaveFile(**self.parameters)
            model.fit(df, self.Base.output_directory)
        except Exception as err:
            msg = "@Task //%i(%s): "%(self.iblock+1, self.SuperFunction) + type(err).__name__ + ": "+ err.message
            raise TypeError(msg)

        // step5: process
        // step6: send out
        order = [edge[1] for edge in self.Base.graph if edge[0] == self.iblock]
        for token in set(order):
            if token == "filepath":
                val = model.file_path
                self.Base.send[(self.iblock, token)] = [val, order.count(token),
                                                        (self.iblock, token, self.Host, self.Function)]
            else:
                msg = "@Task //%i(%s): asked to send a non valid output token "%s"" % (self.iblock+1,self.SuperFunction,token)
                raise NameError(msg)
        // step7: delete all inputs from memory

After Change

class SaveFile(BASE):
    def fit(self):
        // step1: check inputs
        self.required("df", req=True)
        df = self.inputs["df"].value

        // step2: assign inputs to parameters if necessary (param = @token)
        self.paramFROMinput()

        // step3: check the dimension of input data frame
        // step4: import module and make APIs
        try:
            from cheml.initialization import SaveFile
            model = SaveFile(**self.parameters)
            model.fit(df, self.Base.output_directory)
        except Exception as err:
            msg = "@Task //%i(%s): "%(self.iblock+1, self.Task) + type(err).__name__ + ": "+ err.message
            raise TypeError(msg)

        // step5: process
        // step6: send out
        order = [edge[1] for edge in self.Base.graph if edge[0] == self.iblock]
        for token in set(order):
            if token not in self.outputs:
                msg = "@Task //%i(%s): not a valid output token "%s"" % (self.iblock + 1, self.Task, token)
                raise NameError(msg)
            elif token == "filepath":
                self.set_value(token, model.file_path)
                self.outputs[token].count = order.count(token)
                self.Base.send[(self.iblock, token)] = self.outputs[token]

        // step7: delete all inputs from memory
        del self.inputs

Instances