diff --git a/pkg/cli/predict.go b/pkg/cli/predict.go index 54a5dc1d63..ffced4735b 100644 --- a/pkg/cli/predict.go +++ b/pkg/cli/predict.go @@ -90,6 +90,9 @@ func cmdPredict(cmd *cobra.Command, args []string) error { if buildFast { imageName = config.DockerImageName(projectDir) + if err := image.Build(ctx, cfg, projectDir, imageName, buildSecrets, buildNoCache, buildSeparateWeights, buildUseCudaBaseImage, buildProgressOutput, buildSchemaFile, buildDockerfileFile, DetermineUseCogBaseImage(cmd), buildStrip, buildPrecompile, buildFast, nil, buildLocalImage, dockerCommand); err != nil { + return err + } } else { if imageName, err = image.BuildBase(ctx, cfg, projectDir, buildUseCudaBaseImage, DetermineUseCogBaseImage(cmd), buildProgressOutput); err != nil { return err diff --git a/test-integration/test_integration/test_predict.py b/test-integration/test_integration/test_predict.py index 7965641467..df4a7a03ef 100644 --- a/test-integration/test_integration/test_predict.py +++ b/test-integration/test_integration/test_predict.py @@ -542,3 +542,16 @@ def test_predict_granite_project(docker_image): ) assert result.returncode == 0 assert result.stdout == "2.11.3\n" + + +def test_predict_fast_build(docker_image): + project_dir = Path(__file__).parent / "fixtures/fast-build" + + result = subprocess.run( + ["cog", "predict", "--x-fast", "-i", "s=world"], + cwd=project_dir, + capture_output=True, + text=True, + ) + assert result.returncode == 0 + assert result.stdout == "hello world\n"