Get experiment

include/mlflow/client.hpp

@@ -43,6 +43,7 @@ class MlflowClient {
   Result<std::string> get_experiment_by_name(const std::string &name);
   Result<std::string> delete_experiment(const std::string& experiment_id);
   Result<std::string> restore_experiment(const std::string& experiment_id);
+  Result<std::string> update_experiment(const std::string& experiment_id, const std::string& new_name);
 
 private:
   std::unique_ptr<HttpTransport> transport_;

include/mlflow/experiments.hpp

@@ -27,6 +27,7 @@ class Experiments {
   Result<std::string> get_experiment_by_name(const std::string &name);
   Result<std::string> delete_experiment(const std::string& experiment_id);
   Result<std::string> restore_experiment(const std::string& experiment_id);
+  Result<std::string> update_experiment(const std::string& experiment_id, const std::string& new_name);
 
 private:
   HttpTransport &transport_;

src/client.cpp

@@ -38,6 +38,12 @@ MlflowClient::restore_experiment(const std::string& experiment_id)
   return experiments_sub_.restore_experiment(experiment_id);
 }
 
+Result<std::string>
+MlflowClient::update_experiment(const std::string& experiment_id, const std::string& new_name)
+{
+  return experiments_sub_.update_experiment(experiment_id, new_name);
+}
+
 Result<Run> MlflowClient::create_run(const std::string &experiment_id,
                                      const TimestampMs &start_time) {
   return runs_sub_.create_run(experiment_id, start_time);

src/experiments.cpp

@@ -127,4 +127,29 @@ Experiments::restore_experiment(const std::string& experiment_id)
   };
 }
 
+Result<std::string>
+Experiments::update_experiment(const std::string& experiment_id, const std::string& new_name)
+{
+  json payload = {{ "experiment_id", experiment_id }, { "new_name", new_name }};
+  auto res = transport_.post("/experiments/update", payload.dump());
+
+  if(res.status_code != 200)
+  {
+    return
+    {
+      .data = "",
+      .success = false,
+      .error_message = ""
+    };
+  }
+
+  auto res_json = json::parse(res.body);
+  return
+  {
+    .data = "",
+    .success = true,
+    .error_message = ""
+  };
+}
+
 } // namespace mlflow

tests/test_experiments.cpp

@@ -108,6 +108,21 @@ TEST_F(MlflowCppClientFixture, RestoreExperiment)
   ASSERT_TRUE(del_exp_res.data.empty());
 
   auto res = client.restore_experiment(exp_res.data);
+  ASSERT_TRUE(res.success);
+  ASSERT_TRUE(res.data.empty());
+}
+
+TEST_F(MlflowCppClientFixture, UpdateExperiment)
+{
+  std::string name = unique_name("update_experiment");
+  auto exp_res = client.create_experiment(name);
+
+  ASSERT_TRUE(exp_res.success);
+  EXPECT_FALSE(exp_res.data.empty());
+
+  std::string new_name = unique_name("new_experiment");
+  auto res = client.update_experiment(exp_res.data, new_name);
+
   ASSERT_TRUE(res.success);
   ASSERT_TRUE(res.data.empty());
 }