Skip analysing packages with @ (#2416)

* Do not attempt to find package name / version of
packages defined by a remote wheel

Author: 8W9aG

pkg/docker/pipeline_push.go

@@ -212,6 +212,9 @@ func validateRequirements(projectDir string, client *http.Client, cfg *config.Co
 				found = true
 				break
 			}
+			if strings.Contains(pipelineRequirement, "@") {
+				continue
+			}
 			pipelinePackage, pipelineVersion, _, _, err := requirements.SplitPinnedPythonRequirement(pipelineRequirement)
 			if err != nil {
 				return err

pkg/docker/pipeline_push_test.go

@@ -352,3 +352,97 @@ func TestPipelinePushSuccessWithAlphaPatch(t *testing.T) {
 	err = PipelinePush(t.Context(), "r8.im/user/test", dir, apiClient, client, cfg)
 	require.NoError(t, err)
 }
+
+func TestPipelinePushSuccessWithURLInstallPath(t *testing.T) {
+	// Setup mock web server for cog.replicate.com (token exchange)
+	webServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Path {
+		case "/api/token/user":
+			// Mock token exchange response
+			//nolint:gosec
+			tokenResponse := `{
+				"keys": {
+					"cog": {
+						"key": "test-api-token",
+						"expires_at": "2024-12-31T23:59:59Z"
+					}
+				}
+			}`
+			w.WriteHeader(http.StatusOK)
+			w.Write([]byte(tokenResponse))
+		default:
+			w.WriteHeader(http.StatusNotFound)
+		}
+	}))
+	defer webServer.Close()
+
+	// Setup mock API server for api.replicate.com (version and release endpoints)
+	apiServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Path {
+		case "/v1/models/user/test/versions":
+			// Mock version creation response
+			versionResponse := `{"id": "test-version-id"}`
+			w.WriteHeader(http.StatusCreated)
+			w.Write([]byte(versionResponse))
+		case "/v1/models/user/test/releases":
+			// Mock release creation response - empty body with 204 status
+			w.WriteHeader(http.StatusNoContent)
+		default:
+			w.WriteHeader(http.StatusNotFound)
+		}
+	}))
+	defer apiServer.Close()
+
+	cdnServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Path {
+		case "/requirements.txt":
+			// Mock requirements.txt response
+			requirementsResponse := "mycustompackage==1.1.0b2\ncoglet @ https://github.com/replicate/cog-runtime/releases/download/v0.1.0-alpha29/coglet-0.1.0a29-py3-none-any.whl"
+			w.Header().Add(EtagHeader, "a")
+			w.WriteHeader(http.StatusOK)
+			w.Write([]byte(requirementsResponse))
+		default:
+			w.WriteHeader(http.StatusNotFound)
+		}
+	}))
+	defer cdnServer.Close()
+
+	webURL, err := url.Parse(webServer.URL)
+	require.NoError(t, err)
+	apiURL, err := url.Parse(apiServer.URL)
+	require.NoError(t, err)
+	cdnURL, err := url.Parse(cdnServer.URL)
+	require.NoError(t, err)
+
+	t.Setenv(env.SchemeEnvVarName, webURL.Scheme)
+	t.Setenv(env.WebHostEnvVarName, webURL.Host)
+	t.Setenv(env.APIHostEnvVarName, apiURL.Host)
+	t.Setenv(env.PipelinesRuntimeHostEnvVarName, cdnURL.Host)
+
+	dir := t.TempDir()
+
+	// Create mock predict
+	predictPyPath := filepath.Join(dir, "predict.py")
+	handle, err := os.Create(predictPyPath)
+	require.NoError(t, err)
+	handle.WriteString("import cog")
+	handle.Close()
+	dockertest.MockCogConfig = "{\"build\":{\"python_version\":\"3.12\",\"python_packages\":[\"torch==2.5.0\",\"mycustompackage>=1.0\"],\"system_packages\":[\"git\"]},\"image\":\"test\",\"predict\":\"" + predictPyPath + ":Predictor\"}"
+
+	// Setup mock command
+	command := dockertest.NewMockCommand()
+	client, err := cogHttp.ProvideHTTPClient(t.Context(), command)
+	require.NoError(t, err)
+	webClient := web.NewClient(command, client)
+	apiClient := api.NewClient(command, client, webClient)
+
+	cfg := config.DefaultConfig()
+	requirementsPath := filepath.Join(dir, "requirements.txt")
+	handle, err = os.Create(requirementsPath)
+	require.NoError(t, err)
+	handle.WriteString("mycustompackage>=1.0")
+	handle.Close()
+	cfg.Build.PythonRequirements = filepath.Base(requirementsPath)
+	err = PipelinePush(t.Context(), "r8.im/user/test", dir, apiClient, client, cfg)
+	require.NoError(t, err)
+}