From 1e841b32490c6ed5fef52b17abfd3abec1037f75 Mon Sep 17 00:00:00 2001 From: Karel Suta Date: Thu, 1 Feb 2024 15:37:09 +0100 Subject: [PATCH] Add support for providing PyPI connection properties --- support/environment.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/support/environment.go b/support/environment.go index 7645b3d..84e1e23 100644 --- a/support/environment.go +++ b/support/environment.go @@ -47,6 +47,10 @@ const ( // URL for downloading MNIST dataset mnistDatasetURL = "MNIST_DATASET_URL" + + // URL for PiPI index containing all the required test Python packages + pipIndexURL = "PIP_INDEX_URL" + pipTrustedHost = "PIP_TRUSTED_HOST" ) type ClusterType string @@ -117,6 +121,14 @@ func GetMnistDatasetURL() string { return lookupEnvOrDefault(mnistDatasetURL, "http://yann.lecun.com/exdb/mnist/") } +func GetPipIndexURL() string { + return lookupEnvOrDefault(pipIndexURL, "https://pypi.python.org/simple") +} + +func GetPipTrustedHost() string { + return lookupEnvOrDefault(pipTrustedHost, "") +} + func lookupEnvOrDefault(key, value string) string { if v, ok := os.LookupEnv(key); ok { return v