Skip to content

Commit

Permalink
release upper bound for torch, torchvision
Browse files Browse the repository at this point in the history
  • Loading branch information
dev-rinchin committed Dec 12, 2024
1 parent e362126 commit 86bbe76
Showing 1 changed file with 2 additions and 8 deletions.
10 changes: 2 additions & 8 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -54,10 +54,7 @@ lightgbm = [
catboost = ">=0.26.1"
xgboost = "^2.0.0"
optuna = "*"
torch = [
{version = ">=1.9.0, <=2.0.0", python = "<3.11"},
{version = "*", python = ">=3.11"},
]
torch = ">=1.9.0"
holidays = "*"
statsmodels = "<=0.14.0"
networkx = "*"
Expand All @@ -82,10 +79,7 @@ albumentations = {version = "<=1.0.3", optional = true}
timm = {version = ">=0.9.0", optional = true}
opencv-python = {version = "<=4.8.0.74", optional = true}
PyWavelets = {version = "*", optional = true}
torchvision = [
{version = "<=0.14.0", python = "<3.11", optional = true},
{version = "*", python = ">=3.11", optional = true},
]
torchvision = {version = "*", optional = true}

# AFG
featuretools = {version = ">=1.11.1", optional = true}
Expand Down

0 comments on commit 86bbe76

Please sign in to comment.