1 files changed, 19 insertions, 0 deletions
diff --git a/lib/api/entities/ml/mlflow/update_run.rb b/lib/api/entities/ml/mlflow/update_run.rb
new file mode 100644
index 00000000000..5acdaab0e33
--- /dev/null
+++ b/lib/api/entities/ml/mlflow/update_run.rb
@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module API
+  module Entities
+    module Ml
+      module Mlflow
+        class UpdateRun < Grape::Entity
+          expose :run_info
+
+          private
+
+          def run_info
+            ::API::Entities::Ml::Mlflow::RunInfo.represent object
+          end
+        end
+      end
+    end
+  end
+end