Skip to content

Commit 611a9e2

Browse files
author
Muhammad Naufal Andika Natsir Putra
committed
chore: fix golangci-lint related issues
1 parent ddd1b85 commit 611a9e2

File tree

3 files changed

+55
-75
lines changed

3 files changed

+55
-75
lines changed

api/turing/batch/ensembling/runner.go

Lines changed: 1 addition & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -287,28 +287,9 @@ func (r *ensemblingJobRunner) processOneEnsemblingJob(ensemblingJob *models.Ense
287287
return
288288
}
289289

290-
// Get ensembler
291-
ensembler, err := r.ensemblersService.FindByID(
292-
ensemblingJob.EnsemblerID,
293-
service.EnsemblersFindByIDOptions{ProjectID: &ensemblingJob.ProjectID},
294-
)
295-
if err != nil {
296-
r.saveStatusOrTerminate(ensemblingJob, mlpProject, models.JobPending, err.Error(), true)
297-
return
298-
}
299-
300-
// Get base image tag
301-
var baseImageTag string
302-
if pyfuncEnsembler, ok := ensembler.(*models.PyFuncEnsembler); ok {
303-
baseImageTag = pyfuncEnsembler.PythonVersion
304-
} else {
305-
r.saveStatusOrTerminate(ensemblingJob, mlpProject, models.JobPending, err.Error(), true)
306-
return
307-
}
308-
309290
// Build Image
310291
labels := r.buildLabels(ensemblingJob, mlpProject)
311-
imageRef, imageBuildErr := r.buildImage(ensemblingJob, mlpProject, labels, baseImageTag)
292+
imageRef, imageBuildErr := r.buildImage(ensemblingJob, mlpProject, labels)
312293

313294
if imageBuildErr != nil {
314295
// Here unfortunately we have to wait till the image building process has
@@ -433,7 +414,6 @@ func (r *ensemblingJobRunner) buildImage(
433414
ensemblingJob *models.EnsemblingJob,
434415
mlpProject *mlp.Project,
435416
buildLabels map[string]string,
436-
baseImageTag string,
437417
) (string, error) {
438418
request := imagebuilder.BuildImageRequest{
439419
ProjectName: mlpProject.Name,

api/turing/imagebuilder/imagebuilder.go

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ func (ib *imageBuilder) BuildImage(request BuildImageRequest) (string, error) {
174174
return imageRef, nil
175175
}
176176

177-
hashedModelDependenciesUrl, err := ib.getHashedModelDependenciesUrl(context.Background(), request.ArtifactURI)
177+
hashedModelDependenciesURL, err := ib.gethashedModelDependenciesURL(context.Background(), request.ArtifactURI)
178178
if err != nil {
179179
log.Errorf("unable to get model dependencies url: %v", err)
180180
return "", err
@@ -200,7 +200,7 @@ func (ib *imageBuilder) BuildImage(request BuildImageRequest) (string, error) {
200200
}
201201

202202
job, err = ib.createKanikoJob(kanikoJobName, imageRef, request.ArtifactURI, request.BuildLabels,
203-
request.EnsemblerFolder, hashedModelDependenciesUrl)
203+
request.EnsemblerFolder, hashedModelDependenciesURL)
204204
if err != nil {
205205
log.Errorf("unable to build image %s, error: %v", imageRef, err)
206206
return "", ErrUnableToBuildImage
@@ -221,7 +221,7 @@ func (ib *imageBuilder) BuildImage(request BuildImageRequest) (string, error) {
221221
}
222222

223223
job, err = ib.createKanikoJob(kanikoJobName, imageRef, request.ArtifactURI, request.BuildLabels,
224-
request.EnsemblerFolder, hashedModelDependenciesUrl)
224+
request.EnsemblerFolder, hashedModelDependenciesURL)
225225
if err != nil {
226226
log.Errorf("unable to build image %s, error: %v", imageRef, err)
227227
return "", ErrUnableToBuildImage
@@ -290,7 +290,7 @@ func (ib *imageBuilder) createKanikoJob(
290290
artifactURI string,
291291
buildLabels map[string]string,
292292
ensemblerFolder string,
293-
hashedModelDependenciesUrl string,
293+
hashedModelDependenciesURL string,
294294
) (*apibatchv1.Job, error) {
295295
splitURI := strings.Split(artifactURI, "/")
296296
folderName := fmt.Sprintf("%s/%s", splitURI[len(splitURI)-1], ensemblerFolder)
@@ -302,7 +302,7 @@ func (ib *imageBuilder) createKanikoJob(
302302
fmt.Sprintf("--build-arg=BASE_IMAGE=%s", ib.imageBuildingConfig.BaseImage),
303303
fmt.Sprintf("--build-arg=MLFLOW_ARTIFACT_STORAGE_TYPE=%s", ib.artifactServiceType),
304304
fmt.Sprintf("--build-arg=FOLDER_NAME=%s", folderName),
305-
fmt.Sprintf("--build-arg=MODEL_DEPENDENCIES_URL=%s", hashedModelDependenciesUrl),
305+
fmt.Sprintf("--build-arg=MODEL_DEPENDENCIES_URL=%s", hashedModelDependenciesURL),
306306
fmt.Sprintf("--destination=%s", imageRef),
307307
}
308308

@@ -667,20 +667,20 @@ func parseJobConditions(jobConditions []apibatchv1.JobCondition) (string, error)
667667
return jobTable, err
668668
}
669669

670-
// getHashedModelDependenciesUrl stores dependency to storage using it's content hashed name as filename.
670+
// gethashedModelDependenciesURL stores dependency to storage using it's content hashed name as filename.
671671
// if the artifact is recreated with different id but has the same dependency content the URL for the
672672
// stored dependency will still be the same. This ensure we can use Docker layer caching mechanism for
673673
// the built dependency even for different artifact id given the dependency content is not changed.
674-
func (ib *imageBuilder) getHashedModelDependenciesUrl(ctx context.Context, artifactURI string) (string, error) {
674+
func (ib *imageBuilder) gethashedModelDependenciesURL(ctx context.Context, artifactURI string) (string, error) {
675675
artifactURL, err := ib.artifactService.ParseURL(artifactURI)
676676
if err != nil {
677677
return "", err
678678
}
679679

680680
urlSchema := ib.artifactService.GetURLScheme()
681-
condaEnvUrl := fmt.Sprintf("%s://%s/%s/ensembler/conda.yaml", urlSchema, artifactURL.Bucket, artifactURL.Object)
681+
condaEnvURL := fmt.Sprintf("%s://%s/%s/ensembler/conda.yaml", urlSchema, artifactURL.Bucket, artifactURL.Object)
682682

683-
condaEnv, err := ib.artifactService.ReadArtifact(ctx, condaEnvUrl)
683+
condaEnv, err := ib.artifactService.ReadArtifact(ctx, condaEnvURL)
684684
if err != nil {
685685
return "", err
686686
}

0 commit comments

Comments
 (0)