Skip to content

Commit a6401ae

Browse files
python312Packages.torch-geometric: skip failing test on darwin (#350200)
2 parents e45b2e5 + 34395ae commit a6401ae

File tree

1 file changed

+22
-15
lines changed
  • pkgs/development/python-modules/torch-geometric

1 file changed

+22
-15
lines changed

pkgs/development/python-modules/torch-geometric/default.nix

Lines changed: 22 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
{
22
lib,
3+
stdenv,
34
buildPythonPackage,
45
fetchFromGitHub,
56

@@ -153,21 +154,27 @@ buildPythonPackage rec {
153154
export HOME=$(mktemp -d)
154155
'';
155156

156-
disabledTests = [
157-
# TODO: try to re-enable when triton will have been updated to 3.0
158-
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
159-
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
160-
"test_compile_hetero_conv_graph_breaks"
161-
"test_compile_multi_aggr_sage_conv"
162-
163-
# RuntimeError: addmm: computation on CPU is not implemented for SparseCsr + SparseCsr @ SparseCsr without MKL.
164-
# PyTorch built with MKL has better support for addmm with sparse CPU tensors.
165-
"test_asap"
166-
"test_graph_unet"
167-
168-
# AttributeError: type object 'Any' has no attribute '_name'
169-
"test_type_repr"
170-
];
157+
disabledTests =
158+
[
159+
# TODO: try to re-enable when triton will have been updated to 3.0
160+
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
161+
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
162+
"test_compile_hetero_conv_graph_breaks"
163+
"test_compile_multi_aggr_sage_conv"
164+
165+
# RuntimeError: addmm: computation on CPU is not implemented for SparseCsr + SparseCsr @ SparseCsr without MKL.
166+
# PyTorch built with MKL has better support for addmm with sparse CPU tensors.
167+
"test_asap"
168+
"test_graph_unet"
169+
170+
# AttributeError: type object 'Any' has no attribute '_name'
171+
"test_type_repr"
172+
]
173+
++ lib.optionals stdenv.hostPlatform.isDarwin [
174+
# This test uses `torch.jit` which might not be working on darwin:
175+
# RuntimeError: required keyword attribute 'value' has the wrong type
176+
"test_traceable_my_conv_with_self_loops"
177+
];
171178

172179
meta = {
173180
description = "Graph Neural Network Library for PyTorch";

0 commit comments

Comments
 (0)