6c71c2ee1a0d324b0da26950d1cb85a1a0b2f5e4,mlflow/store/model_registry/sqlalchemy_store.py,SqlAlchemyStore,delete_model_version,#SqlAlchemyStore#Any#,289

Before Change

        
        with self.ManagedSessionMaker() as session:
            sql_model_version = self._get_sql_model_version(session, model_version)
            session.delete(sql_model_version)

    def get_model_version_details(self, model_version):
        
        :param model_version: :py:class:`mlflow.entities.model_registry.ModelVersion` object.

After Change

        :return: None
        
        with self.ManagedSessionMaker() as session:
            updated_time = now()
            sql_model_version = self._get_sql_model_version(session, model_version)
            sql_registered_model = sql_model_version.registered_model
            sql_registered_model.last_updated_time = updated_time
            sql_model_version.current_stage = STAGE_DELETED_INTERNAL

Instances