From 3571571119827017b1e289f99930ace36d8e383a Mon Sep 17 00:00:00 2001 From: Ben Firshman Date: Tue, 20 Sep 2022 09:15:47 -0700 Subject: [PATCH] wip Signed-off-by: Ben Firshman --- tools/generate_compatibility_matrices/main.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tools/generate_compatibility_matrices/main.go b/tools/generate_compatibility_matrices/main.go index aecfc69ed9..75d1e6d6b5 100644 --- a/tools/generate_compatibility_matrices/main.go +++ b/tools/generate_compatibility_matrices/main.go @@ -170,6 +170,25 @@ type pypiResponse struct { } `json:"info"` } +func fetchTorchVersions() ([]config.TorchCompatibility, error) { + compats := []config.TorchCompatibility{} + linkRe := regexp.MustCompile(``) + + url := "https://download.pytorch.org/whl/torch_stable.html" + resp, err := soup.Get(url) + if err != nil { + return nil, fmt.Errorf("Failed to download %s: %w", url, err) + } + doc := soup.HTMLParse(resp) + links := doc.FindAll("a") + for _, link := range links { + href := link.Attrs()["href"] + + } + + return compats, nil +} + // Need to get latest versions because PyTorch doesn't specify versions for latest func fetchDefaultVersions() (map[string]string, error) { fmt.Println("Fetching default versions...")