lib/api/entities/ml/mlflow/experiment.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

# frozen_string_literal: true

module API
  module Entities
    module Ml
      module Mlflow
        class Experiment < Grape::Entity
          expose(:experiment_id) { |experiment| experiment.iid.to_s }
          expose :name
          expose(:lifecycle_stage) { |experiment| experiment.deleted_on? ? 'deleted' : 'active' }
          expose(:artifact_location) { |experiment| 'not_implemented' }
          expose :metadata, as: :tags, using: KeyValue
        end
      end
    end
  end
end