29628cab50c7b84ed64776fa02f6eb6e29a55569,dataset/pipeline.py,Pipeline,_exec_import_model,#Pipeline#Any#Any#,715

Before Change

    def _exec_import_model(self, _, action):
        model_name = self._get_value(action["model_name"])
        if ModelDirectory.find_model_by_name(model_name, pipeline=self) is None:
            with self._models_lock:
                if ModelDirectory.find_model_by_name(model_name, pipeline=self) is None:
                    if isinstance(action["source"], Pipeline):
                        ModelDirectory.import_model_from(model_name, action["source"], self)
                    else:
                        ModelDirectory.import_model(model_name, action["source"], self)

    def train_model(self, name, make_data=None, save_to=None, mode="w", *args, **kwargs):
         Train a model

        Parameters

After Change

    def _exec_import_model(self, batch, action):
        model_name = self._get_value(action["model_name"], batch=batch)
        source_name = self._get_value(action["source_name"], batch=batch)
        source = self._get_value(action["source"], batch=batch)
        self.models.import_model(source_name, source, model_name)

    def train_model(self, name, make_data=None, save_to=None, mode="w", *args, **kwargs):

Instances