diff options
Diffstat (limited to 'app/presenters/ml/model_version_presenter.rb')
-rw-r--r-- | app/presenters/ml/model_version_presenter.rb | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/app/presenters/ml/model_version_presenter.rb b/app/presenters/ml/model_version_presenter.rb new file mode 100644 index 00000000000..210b213ca2a --- /dev/null +++ b/app/presenters/ml/model_version_presenter.rb @@ -0,0 +1,25 @@ +# frozen_string_literal: true + +module Ml + class ModelVersionPresenter < Gitlab::View::Presenter::Delegated + presents ::Ml::ModelVersion, as: :model_version + + def display_name + "#{model_version.model.name} / #{model_version.version}" + end + + def path + project_ml_model_version_path( + model_version.model.project, + model_version.model, + model_version + ) + end + + def package_path + return unless model_version.package_id.present? + + project_package_path(model_version.project, model_version.package_id) + end + end +end |