1 files changed, 5 insertions, 7 deletions
diff --git a/lib/api/entities/ml/mlflow/run.rb b/lib/api/entities/ml/mlflow/run.rb
index 01d85e8862b..10e2434521d 100644
--- a/lib/api/entities/ml/mlflow/run.rb
+++ b/lib/api/entities/ml/mlflow/run.rb
@@ -5,13 +5,11 @@ module API
     module Ml
       module Mlflow
         class Run < Grape::Entity
-          expose :run do
-            expose :itself, using: RunInfo, as: :info
-            expose :data do
-              expose :metrics, using: Metric
-              expose :params, using: KeyValue
-              expose :metadata, as: :tags, using: KeyValue
-            end
+          expose :itself, using: RunInfo, as: :info
+          expose :data do
+            expose :metrics, using: Metric
+            expose :params, using: KeyValue
+            expose :metadata, as: :tags, using: KeyValue
           end
         end
       end