Skip to content

Commit

Permalink
Just add all the packages that transitively depend on torch for now (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
blast-hardcheese authored Jan 28, 2025
1 parent 81b5361 commit 01b04d8
Show file tree
Hide file tree
Showing 2 changed files with 1,120 additions and 3 deletions.
4 changes: 1 addition & 3 deletions internal/backends/python/python.go
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Package python provides backends for Python 2 and 3 using Poetry and pip.
// Package python provides backends for Python 3 using uv, poetry and pip.
package python

import (
Expand All @@ -23,8 +23,6 @@ import (

var normalizationPattern = regexp.MustCompile(`[-_.]+`)

var torchOverrides = []string{"torch", "torchvision"}

// this generates a mapping of pypi packages <-> modules
// moduleToPypiPackage pypiPackageToModules are provided
// pypiEntryInfoResponse is a wrapper around pypiEntryInfo
Expand Down
Loading

0 comments on commit 01b04d8

Please sign in to comment.