app/presenters/ml/model_presenter.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

# frozen_string_literal: true

module Ml
  class ModelPresenter < Gitlab::View::Presenter::Delegated
    presents ::Ml::Model, as: :model

    def latest_version_name
      latest_version&.version
    end

    def version_count
      return model.version_count if model.respond_to?(:version_count)

      model.versions.size
    end

    def latest_package_path
      latest_version&.package_path
    end

    def latest_version_path
      latest_version&.path
    end

    def path
      project_ml_model_path(model.project, model.id)
    end

    private

    def latest_version
      model.latest_version&.present
    end
  end
end