happy now, pip?

neggles · Feb 26, 2024 · 060ef61 · 060ef61
1 parent ab40549
commit 060ef61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -64,7 +64,7 @@ function torchName {
 # torch version to torch name
 function torchSpec {
   params = [version]
-  result = regex_replace(version, "^(\\d+)\\.(\\d+).*", "torch~=$1.$2.$3")
+  result = regex_replace(version, "^(\\d+)\\.(\\d+).*", "torch>=$1.$2.$3,==$1.$2.*")
 }
 
 # build a tag for an image from this repo