chore: drop coglet-alpha selection

Map coglet-alpha and coglet wheel selections to cog-dataclass and remove the pinned coglet-alpha install path.
Update dockerfile generators, wheel parsing, CI runtime matrix, and integration-test conditions to reflect the removal.

Author: mfainberg-cf

.github/workflows/ci.yaml

@@ -210,7 +210,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        runtime: [cog, cog-dataclass, coglet-alpha, cog-rust, cog-dataclass-rust]
+        runtime: [cog, cog-dataclass, cog-rust, cog-dataclass-rust]
     steps:
       - uses: actions/checkout@v6
         with:
@@ -246,9 +246,6 @@ jobs:
             cog-dataclass)
               echo "COG_WHEEL=cog-dataclass" >> $GITHUB_ENV
               ;;
-            coglet-alpha)
-              echo "COG_WHEEL=coglet-alpha" >> $GITHUB_ENV
-              ;;
             cog-rust)
               RUST_WHEEL=$(ls dist/coglet-*-cp310-abi3-*.whl | head -1)
               echo "Found Rust coglet wheel: $RUST_WHEEL"

architecture/05-build-system.md

@@ -195,7 +195,7 @@ The `COG_WHEEL` environment variable controls which wheel is installed:
 |-------|--------|
 | (unset) | Embedded `cog` wheel (default) |
 | `cog` | Embedded `cog` wheel |
-| `coglet` / `coglet-alpha` | Experimental runtime (temporary compatibility) |
+| `coglet` | Experimental runtime (temporary compatibility) |
 | `https://...` | Download wheel from URL |
 | `/path/to/file.whl` | Use local wheel file |
 

integration-tests/concurrent/concurrent_test.go

@@ -31,13 +31,6 @@ func TestConcurrentPredictions(t *testing.T) {
 		t.Skip("skipping slow test in short mode")
 	}
 
-	// Skip for coglet-alpha runtime - it has a known bug with concurrent async predictions
-	// that causes predictions to return empty output. This is in the released version and
-	// cannot be fixed without a new release.
-	if cogWheel := os.Getenv("COG_WHEEL"); cogWheel == "coglet-alpha" {
-		t.Skip("skipping: coglet-alpha has a known bug with concurrent async predictions")
-	}
-
 	// Create a temp directory for our test project
 	tmpDir, err := os.MkdirTemp("", "cog-concurrent-test-*")
 	if err != nil {

integration-tests/suite_test.go

@@ -29,7 +29,7 @@ func TestIntegration(t *testing.T) {
 // condition provides custom conditions for testscript.
 // Supported conditions:
 //   - linux/linux_amd64/amd64: platform guards for specialized tests.
-//   - coglet_alpha: true when COG_WHEEL=coglet-alpha (old Go coglet)
+//   - coglet_alpha: deprecated (always false)
 //   - cog_dataclass: true when COG_WHEEL=cog-dataclass (Python 3.10+ only)
 //   - cog_rust: true when COG_WHEEL=cog and COGLET_RUST_WHEEL is set
 //   - cog_dataclass_rust: true when COG_WHEEL=cog-dataclass and COGLET_RUST_WHEEL is set
@@ -55,7 +55,7 @@ func condition(cond string) (bool, error) {
 	case "linux_amd64":
 		value = runtime.GOOS == "linux" && runtime.GOARCH == "amd64"
 	case "coglet_alpha":
-		value = cogWheel == "coglet-alpha"
+		value = false
 	case "cog_dataclass":
 		value = cogWheel == "cog-dataclass"
 	case "cog_rust":

pkg/dockerfile/fast_generator.go

@@ -216,7 +216,7 @@ func (g *FastGenerator) generateMonobase(lines []string, tmpDir string) ([]strin
 	var envs []string
 
 	envs = append(envs, []string{
-		"ENV R8_COG_VERSION=" + PinnedCogletURL,
+		"ENV R8_COG_VERSION=cog-dataclass",
 	}...)
 
 	torchVersion, ok := g.Config.TorchVersion()

pkg/dockerfile/standard_generator.go

@@ -47,7 +47,6 @@ const StripDebugSymbolsCommand = "find / -type f -name \"*python*.so\" -not -nam
 const CFlags = "ENV CFLAGS=\"-O3 -funroll-loops -fno-strict-aliasing -flto -S\""
 const PrecompilePythonCommand = "RUN find / -type f -name \"*.py[co]\" -delete && find / -type f -name \"*.py\" -exec touch -t 197001010000 {} \\; && find / -type f -name \"*.py\" -printf \"%h\\n\" | sort -u | /usr/bin/python3 -m compileall --invalidation-mode timestamp -o 2 -j 0"
 const STANDARD_GENERATOR_NAME = "STANDARD_GENERATOR"
-const PinnedCogletURL = "https://github.com/replicate/cog-runtime/releases/download/v0.1.0-beta10/coglet-0.1.0b10-py3-none-any.whl" // Pinned coglet URL to avoid API dependency
 
 type StandardGenerator struct {
 	Config *config.Config
@@ -475,8 +474,6 @@ func (g *StandardGenerator) installCog() (string, error) {
 	switch wheelConfig.Source {
 	case wheels.WheelSourceCog:
 		installLines, err = g.installEmbeddedCogWheel()
-	case wheels.WheelSourceCogletAlpha:
-		installLines, err = g.installCogletAlpha()
 	case wheels.WheelSourceCogDataclass:
 		installLines, err = g.installEmbeddedCogDataclassWheel()
 	case wheels.WheelSourceURL:
@@ -493,18 +490,13 @@ func (g *StandardGenerator) installCog() (string, error) {
 
 	// Optionally install Rust coglet wheel alongside cog
 	// This allows testing the Rust HTTP server implementation
-	// WARNING: Cannot install Rust coglet when using coglet-alpha
 	if rustWheelPath := os.Getenv("COGLET_RUST_WHEEL"); rustWheelPath != "" {
-		if wheelConfig != nil && wheelConfig.Source == wheels.WheelSourceCogletAlpha {
-			fmt.Fprintf(os.Stderr, "WARNING: COGLET_RUST_WHEEL is set but COG_WHEEL=coglet-alpha. Rust coglet cannot be installed alongside coglet-alpha (Go implementation). Skipping Rust coglet installation.\n")
-		} else {
-			rustInstall, err := g.installRustCogletWheel(rustWheelPath)
-			if err != nil {
-				return "", fmt.Errorf("failed to install Rust coglet wheel: %w", err)
-			}
-			if rustInstall != "" {
-				installLines += "\n" + rustInstall
-			}
+		rustInstall, err := g.installRustCogletWheel(rustWheelPath)
+		if err != nil {
+			return "", fmt.Errorf("failed to install Rust coglet wheel: %w", err)
+		}
+		if rustInstall != "" {
+			installLines += "\n" + rustInstall
 		}
 	}
 
@@ -545,23 +537,6 @@ func (g *StandardGenerator) installEmbeddedCogDataclassWheel() (string, error) {
 	return strings.Join(lines, "\n"), nil
 }
 
-// installCogletAlpha installs coglet from the pinned URL (default for cog_runtime: true)
-func (g *StandardGenerator) installCogletAlpha() (string, error) {
-	// We need fast-* compliant Python version to reconstruct coglet venv PYTHONPATH
-	if !CheckMajorMinorOnly(g.Config.Build.PythonVersion) {
-		return "", fmt.Errorf("Python version must be <major>.<minor>")
-	}
-	cmds := []string{
-		"ENV R8_COG_VERSION=coglet",
-		"ENV R8_PYTHON_VERSION=" + g.Config.Build.PythonVersion,
-		// Uninstall cog first to avoid conflicts with coglet's cog shim package.
-		// Some base images (e.g. r8.im/cog-base) have cog pre-installed, which conflicts
-		// with coglet's cog compatibility shim that provides the same module paths.
-		"RUN pip uninstall -y cog 2>/dev/null || true && pip install " + PinnedCogletURL,
-	}
-	return strings.Join(cmds, "\n"), nil
-}
-
 // installWheelFromURL installs a wheel from a URL (when COG_WHEEL=https://...)
 func (g *StandardGenerator) installWheelFromURL(url string) (string, error) {
 	// Set coglet env vars if this looks like a coglet wheel

pkg/dockerfile/standard_generator_test.go

@@ -925,7 +925,7 @@ predict: predict.py:Predictor
 }
 
 func TestCOGWheelDefaultCogRuntimeTrue(t *testing.T) {
-	// Default behavior with cog_runtime: true should use coglet-alpha (PinnedCogletURL)
+	// Default behavior with cog_runtime: true should use cog-dataclass
 	tmpDir := t.TempDir()
 
 	yaml := `
@@ -947,10 +947,11 @@ predict: predict.py:Predictor
 	_, actual, _, err := gen.GenerateModelBaseWithSeparateWeights(t.Context(), "r8.im/replicate/cog-test")
 	require.NoError(t, err)
 
-	// Should contain coglet-alpha/PinnedCogletURL install with cog uninstall prefix
-	require.Contains(t, actual, "RUN pip uninstall -y cog 2>/dev/null || true && pip install "+PinnedCogletURL)
-	require.Contains(t, actual, "ENV R8_COG_VERSION=coglet")
-	require.Contains(t, actual, "ENV R8_PYTHON_VERSION=3.11")
+	// Should contain the embedded cog-dataclass wheel install
+	require.Contains(t, actual, "/tmp/cog_dataclass-")
+	require.Contains(t, actual, ".whl")
+	require.NotContains(t, actual, "'pydantic>=1.9,<3'")
+	require.NotContains(t, actual, "R8_COG_VERSION=coglet")
 }
 
 func TestCOGWheelEnvCog(t *testing.T) {
@@ -987,7 +988,7 @@ predict: predict.py:Predictor
 }
 
 func TestCOGWheelEnvCogletAlpha(t *testing.T) {
-	// COG_WHEEL=coglet-alpha should use PinnedCogletURL even without cog_runtime: true
+	// COG_WHEEL=coglet-alpha should map to cog-dataclass
 	t.Setenv("COG_WHEEL", "coglet-alpha")
 
 	tmpDir := t.TempDir()
@@ -1010,10 +1011,11 @@ predict: predict.py:Predictor
 	_, actual, _, err := gen.GenerateModelBaseWithSeparateWeights(t.Context(), "r8.im/replicate/cog-test")
 	require.NoError(t, err)
 
-	// Should contain coglet-alpha/PinnedCogletURL install with cog uninstall prefix
-	require.Contains(t, actual, "RUN pip uninstall -y cog 2>/dev/null || true && pip install "+PinnedCogletURL)
-	require.Contains(t, actual, "ENV R8_COG_VERSION=coglet")
-	require.Contains(t, actual, "ENV R8_PYTHON_VERSION=3.11")
+	// Should contain the embedded cog-dataclass wheel install
+	require.Contains(t, actual, "/tmp/cog_dataclass-")
+	require.Contains(t, actual, ".whl")
+	require.NotContains(t, actual, "'pydantic>=1.9,<3'")
+	require.NotContains(t, actual, "R8_COG_VERSION=coglet")
 }
 
 func TestCOGWheelEnvURL(t *testing.T) {

pkg/wheels/wheels.go

@@ -80,10 +80,6 @@ type WheelSource int
 const (
 	// WheelSourceCog uses the embedded cog wheel (default when cog_runtime: false)
 	WheelSourceCog WheelSource = iota
-	// WheelSourceCogletEmbedded uses the embedded coglet wheel
-	WheelSourceCogletEmbedded
-	// WheelSourceCogletAlpha uses the PinnedCogletURL (default when cog_runtime: true)
-	WheelSourceCogletAlpha
 	// WheelSourceCogDataclass uses the embedded cog-dataclass wheel (pydantic-less)
 	WheelSourceCogDataclass
 	// WheelSourceURL uses a custom URL
@@ -97,10 +93,6 @@ func (s WheelSource) String() string {
 	switch s {
 	case WheelSourceCog:
 		return "cog"
-	case WheelSourceCogletEmbedded:
-		return "coglet"
-	case WheelSourceCogletAlpha:
-		return "coglet-alpha"
 	case WheelSourceCogDataclass:
 		return "cog-dataclass"
 	case WheelSourceURL:
@@ -128,8 +120,7 @@ const CogWheelEnvVar = "COG_WHEEL"
 // ParseCogWheel parses a COG_WHEEL value and returns the appropriate WheelConfig.
 // Supported values:
 //   - "cog" - Embedded cog wheel
-//   - "coglet" - Embedded coglet wheel
-//   - "coglet-alpha" - PinnedCogletURL
+//   - "coglet" - Deprecated (maps to cog-dataclass)
 //   - "cog-dataclass" - Embedded cog-dataclass wheel (pydantic-less)
 //   - "https://..." or "http://..." - Direct wheel URL
 //   - "/path/to/file.whl" or "./path/to/file.whl" - Local wheel file
@@ -145,9 +136,9 @@ func ParseCogWheel(value string) *WheelConfig {
 	case "cog":
 		return &WheelConfig{Source: WheelSourceCog}
 	case "coglet":
-		return &WheelConfig{Source: WheelSourceCogletEmbedded}
+		return &WheelConfig{Source: WheelSourceCogDataclass}
 	case "coglet-alpha":
-		return &WheelConfig{Source: WheelSourceCogletAlpha}
+		return &WheelConfig{Source: WheelSourceCogDataclass}
 	case "cog-dataclass":
 		return &WheelConfig{Source: WheelSourceCogDataclass}
 	}
@@ -164,17 +155,20 @@ func ParseCogWheel(value string) *WheelConfig {
 // GetWheelConfig returns the WheelConfig based on COG_WHEEL env var and cog_runtime flag.
 // Priority:
 //  1. COG_WHEEL env var (if set, overrides everything)
-//  2. cog_runtime: true -> coglet-alpha (PinnedCogletURL)
+//  2. cog_runtime: true -> cog-dataclass (coglet-alpha removed)
 //  3. cog_runtime: false (default) -> embedded cog wheel
 func GetWheelConfig(cogRuntimeEnabled bool) *WheelConfig {
 	envValue := os.Getenv(CogWheelEnvVar)
+	if strings.EqualFold(envValue, "coglet-alpha") || strings.EqualFold(envValue, "coglet") {
+		return &WheelConfig{Source: WheelSourceCogDataclass}
+	}
 	if config := ParseCogWheel(envValue); config != nil {
 		return config
 	}
 
 	// Default based on cog_runtime flag
 	if cogRuntimeEnabled {
-		return &WheelConfig{Source: WheelSourceCogletAlpha}
+		return &WheelConfig{Source: WheelSourceCogDataclass}
 	}
 	return &WheelConfig{Source: WheelSourceCog}
 }

pkg/wheels/wheels_test.go

@@ -20,7 +20,6 @@ func TestWheelSourceString(t *testing.T) {
 		expected string
 	}{
 		{WheelSourceCog, "cog"},
-		{WheelSourceCogletAlpha, "coglet-alpha"},
 		{WheelSourceCogDataclass, "cog-dataclass"},
 		{WheelSourceURL, "url"},
 		{WheelSourceFile, "file"},
@@ -68,15 +67,20 @@ func TestParseCogWheel(t *testing.T) {
 			input:    "Cog",
 			expected: &WheelConfig{Source: WheelSourceCog},
 		},
+		{
+			name:     "coglet keyword",
+			input:    "coglet",
+			expected: &WheelConfig{Source: WheelSourceCogDataclass},
+		},
 		{
 			name:     "coglet-alpha keyword",
 			input:    "coglet-alpha",
-			expected: &WheelConfig{Source: WheelSourceCogletAlpha},
+			expected: &WheelConfig{Source: WheelSourceCogDataclass},
 		},
 		{
 			name:     "coglet-alpha uppercase",
 			input:    "COGLET-ALPHA",
-			expected: &WheelConfig{Source: WheelSourceCogletAlpha},
+			expected: &WheelConfig{Source: WheelSourceCogDataclass},
 		},
 		{
 			name:     "cog with whitespace",
@@ -186,7 +190,7 @@ func TestGetWheelConfig(t *testing.T) {
 			name:              "default with cog_runtime true",
 			envValue:          "",
 			cogRuntimeEnabled: true,
-			expected:          &WheelConfig{Source: WheelSourceCogletAlpha},
+			expected:          &WheelConfig{Source: WheelSourceCogDataclass},
 		},
 
 		// Env var overrides cog_runtime
@@ -200,7 +204,7 @@ func TestGetWheelConfig(t *testing.T) {
 			name:              "env coglet-alpha overrides cog_runtime false",
 			envValue:          "coglet-alpha",
 			cogRuntimeEnabled: false,
-			expected:          &WheelConfig{Source: WheelSourceCogletAlpha},
+			expected:          &WheelConfig{Source: WheelSourceCogDataclass},
 		},
 		{
 			name:              "env URL overrides cog_runtime",