Remove integration-tests/go.mod, consolidate into root module

.github/workflows/ci.yaml

@@ -476,7 +476,7 @@ jobs:
             kill -KILL -- -$$ 2>/dev/null || true
           }
           trap cleanup INT TERM
-          gotestsum -- -short -timeout 1200s -parallel 5 $(go list ./...) &
+          gotestsum -- -short -timeout 1200s -parallel 5 ./... &
           wait $!
 
   test-rust:
@@ -690,9 +690,7 @@ jobs:
       - uses: actions/setup-go@v6
         with:
           go-version-file: go.mod
-          cache-dependency-path: |
-            go.sum
-            integration-tests/go.sum
+          cache-dependency-path: go.sum
       # gotestsum via Go module proxy (not GitHub Releases) for reliability
       - name: Install gotestsum
         run: go install gotest.tools/gotestsum@v1.13.0
@@ -709,7 +707,6 @@ jobs:
           COG_BINARY: ./cog
           TEST_PARALLEL: 4
           BUILDKIT_PROGRESS: 'quiet'
-        working-directory: integration-tests
         shell: bash
         run: |
           set -euo pipefail
@@ -728,9 +725,11 @@ jobs:
           echo "Running tests matching: $RUN_PATTERN"
 
           gotestsum --format github-actions -- \
+            -tags integration \
             -parallel $TEST_PARALLEL \
             -timeout 30m \
-            -run "TestIntegration/($RUN_PATTERN)/" &
+            -run "TestIntegration/($RUN_PATTERN)/" \
+            ./integration-tests/... &
           wait $!
 
 # =============================================================================

go.mod

@@ -7,6 +7,7 @@ require (
 	github.com/aws/aws-sdk-go-v2 v1.39.4
 	github.com/aws/aws-sdk-go-v2/service/s3 v1.88.7
 	github.com/containerd/errdefs v1.0.0
+	github.com/creack/pty v1.1.24
 	github.com/docker/cli v28.5.2+incompatible
 	github.com/docker/docker v28.5.2+incompatible
 	github.com/docker/go-connections v0.5.0
@@ -22,6 +23,7 @@ require (
 	github.com/opencontainers/image-spec v1.1.1
 	github.com/pkg/errors v0.9.1
 	github.com/replicate/go v0.0.0-20250205165008-b772d7cd506b
+	github.com/rogpeppe/go-internal v1.14.1
 	github.com/sabhiram/go-gitignore v0.0.0-20210923224102-525f6e181f06
 	github.com/spf13/cobra v1.9.1
 	github.com/spf13/pflag v1.0.6
@@ -71,7 +73,6 @@ require (
 	github.com/containerd/ttrpc v1.2.7 // indirect
 	github.com/containerd/typeurl/v2 v2.2.3 // indirect
 	github.com/cpuguy83/dockercfg v0.3.2 // indirect
-	github.com/creack/pty v1.1.24 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 	github.com/distribution/reference v0.6.0 // indirect
 	github.com/dnephin/pflag v1.0.7 // indirect
@@ -125,7 +126,6 @@ require (
 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
-	github.com/rogpeppe/go-internal v1.14.1 // indirect
 	github.com/secure-systems-lab/go-securesystemslib v0.4.0 // indirect
 	github.com/shibumi/go-pathspec v1.3.0 // indirect
 	github.com/shirou/gopsutil/v4 v4.25.2 // indirect

integration-tests/concurrent/concurrent_test.go

@@ -1,3 +1,5 @@
+//go:build integration
+
 package concurrent_test
 
 import (
@@ -39,10 +41,10 @@ func TestConcurrentPredictions(t *testing.T) {
 	defer os.RemoveAll(tmpDir)
 
 	// Write the async-sleep predictor fixture
-	if err := os.WriteFile(filepath.Join(tmpDir, "cog.yaml"), []byte(cogYAML), 0644); err != nil {
+	if err := os.WriteFile(filepath.Join(tmpDir, "cog.yaml"), []byte(cogYAML), 0o644); err != nil {
 		t.Fatalf("failed to write cog.yaml: %v", err)
 	}
-	if err := os.WriteFile(filepath.Join(tmpDir, "predict.py"), []byte(predictPy), 0644); err != nil {
+	if err := os.WriteFile(filepath.Join(tmpDir, "predict.py"), []byte(predictPy), 0o644); err != nil {
 		t.Fatalf("failed to write predict.py: %v", err)
 	}