From 22d15b1895af3c813fb7c2490f68158670108d8c Mon Sep 17 00:00:00 2001
From: Ewe Zi Yi <36802364+deadlycoconuts@users.noreply.github.com>
Date: Thu, 5 Oct 2023 09:33:48 +0800
Subject: [PATCH] Change delete ensembler api response from int to struct
 (#364)

---
 api/turing/api/ensemblers_api.go      | 2 +-
 api/turing/api/ensemblers_api_test.go | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/api/turing/api/ensemblers_api.go b/api/turing/api/ensemblers_api.go
index 5de348504..d13fde43c 100644
--- a/api/turing/api/ensemblers_api.go
+++ b/api/turing/api/ensemblers_api.go
@@ -213,7 +213,7 @@ func (c EnsemblersController) DeleteEnsembler(
 		return Error(httpStatus, "failed to delete the ensembler", err.Error())
 	}
 
-	return Ok(ensembler.GetID())
+	return Ok(map[string]int{"id": int(ensembler.GetID())})
 }
 
 func (c EnsemblersController) checkActiveRouterVersion(options EnsemblersPathOptions) (int, error) {
diff --git a/api/turing/api/ensemblers_api_test.go b/api/turing/api/ensemblers_api_test.go
index 6766ac66f..f045a564f 100644
--- a/api/turing/api/ensemblers_api_test.go
+++ b/api/turing/api/ensemblers_api_test.go
@@ -917,7 +917,7 @@ func TestEnsemblerController_DeleteEnsembler(t *testing.T) {
 				mlflowSvc.On("DeleteExperiment", mock.Anything, "1", true).Return(nil)
 				return mlflowSvc
 			},
-			expected: Ok(models.ID(2)),
+			expected: Ok(map[string]int{"id": 2}),
 		},
 		"success": {
 			vars: RequestVars{
@@ -962,7 +962,7 @@ func TestEnsemblerController_DeleteEnsembler(t *testing.T) {
 				mlflowSvc.On("DeleteExperiment", mock.Anything, "1", true).Return(nil)
 				return mlflowSvc
 			},
-			expected: Ok(models.ID(2)),
+			expected: Ok(map[string]int{"id": 2}),
 		},
 	}
 	for name, tt := range tests {