@@ -34,8 +34,8 @@ hydra-core = "^1.2.0"
34
34
35
35
# Optional dependencies
36
36
monai = { version = " ^1.3.0" , extras = [" itk" ], optional = true }
37
- torch = { version = " ^1.11 .0" , optional = true }
38
- torchvision = { version = " ^0.14 .0" , optional = true }
37
+ torch = { version = " ^2.0 .0" , optional = true }
38
+ torchvision = { version = " ^0.15 .0" , optional = true }
39
39
torchxrayvision = { version = " ^1.2.0" , optional = true }
40
40
xgboost = { version = " ^1.5.2" , optional = true }
41
41
alibi = { version = " ^0.9.4" , optional = true , extras = [" shap" ] }
@@ -52,26 +52,26 @@ xgboost = "^1.5.2"
52
52
optional = true
53
53
54
54
[tool .poetry .group .torch .dependencies ]
55
- torch = " ^1.11 .0"
55
+ torch = " ^2.0 .0"
56
56
57
57
[tool .poetry .group .torchvision ]
58
58
optional = true
59
59
60
60
[tool .poetry .group .torchvision .dependencies ]
61
- torchvision = " ^0.14 .0"
61
+ torchvision = " ^0.15 .0"
62
62
63
63
[tool .poetry .group .torchxrayvision ]
64
64
optional = true
65
65
66
66
[tool .poetry .group .torchxrayvision .dependencies ]
67
- torch = " ^1.11 .0"
67
+ torch = " ^2.0 .0"
68
68
torchxrayvision = " ^1.2.0"
69
69
70
70
[tool .poetry .group .monai ]
71
71
optional = true
72
72
73
73
[tool .poetry .group .monai .dependencies ]
74
- torch = " ^1.11 .0"
74
+ torch = " ^2.0 .0"
75
75
monai = " ^1.1.0"
76
76
77
77
[tool .poetry .group .alibi ]
@@ -86,7 +86,7 @@ optional = true
86
86
87
87
[tool .poetry .group .alibi-detect .dependencies ]
88
88
llvmlite = " ^0.40.0"
89
- torch = " ^1.11 .0"
89
+ torch = " ^2.0 .0"
90
90
alibi-detect = { version = " ^0.12.0" , extras = [" torch" ] }
91
91
92
92
[tool .poetry .group .test ]
0 commit comments