Update api server to work with updated mlflow endpoints

caraml-dev · Nov 14, 2024 · abeee76 · abeee76
1 parent 1c49bd7
commit abeee76
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/api/mlflow/mlflow.go b/api/mlflow/mlflow.go
@@ -61,6 +61,7 @@ func (mlflow *client) doCall(req *request, resp interface{}) error {
 		return err
 	}
 
+	httpReq.Header.Set("Content-Type", "application/json")
 	httpResp, err := mlflow.httpClient.Do(httpReq)
 	if err != nil {
 		return err

diff --git a/api/mlflow/response.go b/api/mlflow/response.go
@@ -39,8 +39,8 @@ type Run struct {
 type Info struct {
 	RunID          string `json:"run_id"`
 	ExperimentID   string `json:"experiment_id"`
-	StartTime      string `json:"start_time"`
-	EndTime        string `json:"end_time"`
+	StartTime      int    `json:"start_time"`
+	EndTime        int    `json:"end_time"`
 	ArtifactURI    string `json:"artifact_uri"`
 	LifecycleStage string `json:"lifecycle_stage"`
 	Status         string `json:"status"`