1 files changed, 4 insertions, 16 deletions
diff --git a/lib/api/entities/ml/mlflow/experiment.rb b/lib/api/entities/ml/mlflow/experiment.rb
index cfe366feaab..54e0fe63985 100644
--- a/lib/api/entities/ml/mlflow/experiment.rb
+++ b/lib/api/entities/ml/mlflow/experiment.rb
@@ -5,22 +5,10 @@ module API
     module Ml
       module Mlflow
         class Experiment < Grape::Entity
-          expose :experiment do
-            expose :experiment_id
-            expose :name
-            expose :lifecycle_stage
-            expose :artifact_location
-          end
-
-          private
-
-          def lifecycle_stage
-            object.deleted_on? ? 'deleted' : 'active'
-          end
-
-          def experiment_id
-            object.iid.to_s
-          end
+          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' }
         end
       end
     end