Skip to content

Commit fc39c4d

Browse files
committed
Fix pytorch-tests python pins
Signed-off-by: Michał Górny <[email protected]>
1 parent 316c31d commit fc39c4d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

recipe/meta.yaml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -359,9 +359,8 @@ outputs:
359359
detect_binary_files_with_prefix: false
360360
test:
361361
requires:
362-
- pytorch {{ version }} cuda*_{{ blas_impl }}*{{ build }} # [megabuild and cuda_compiler_version != "None"]
363-
- pytorch {{ version }} cpu_{{ blas_impl }}*{{ build }} # [megabuild and cuda_compiler_version == "None"]
364-
- {{ pin_subpackage("pytorch", exact=True) }} # [not megabuild]
362+
- pytorch {{ version }} cuda*_{{ blas_impl }}_py{{ CONDA_PY }}*{{ build }} # [cuda_compiler_version != "None"]
363+
- pytorch {{ version }} cpu_{{ blas_impl }}_py{{ CONDA_PY }}*{{ build }} # [cuda_compiler_version == "None"]
365364
- {{ compiler('c') }}
366365
- {{ compiler('cxx') }}
367366
# for torch.compile tests

0 commit comments

Comments
 (0)