python3Packages.pylance: re-enable torch compile tests on darwin
This commit is contained in:
@@ -165,19 +165,6 @@ buildPythonPackage (finalAttrs: {
|
||||
++ lib.optionals stdenv.hostPlatform.isDarwin [
|
||||
# Build hangs after all the tests are run due to a torch subprocess not exiting
|
||||
"test_multiprocess_loading"
|
||||
|
||||
# torch._inductor.exc.InductorError: CppCompileError: C++ compile error
|
||||
# OpenMP support not found
|
||||
# TODO: figure out why this only happens on python 3.13 and not 3.14
|
||||
"test_cosine_distance"
|
||||
"test_ground_truth"
|
||||
"test_index_cast_centroids"
|
||||
"test_index_with_no_centroid_movement"
|
||||
"test_l2_distance"
|
||||
"test_l2_distance_f16_bf16_cpu"
|
||||
"test_pairwise_cosine"
|
||||
"test_torch_index_with_nans"
|
||||
"test_torch_kmeans_nans"
|
||||
]
|
||||
++ lib.optionals (pythonAtLeast "3.14") [
|
||||
# RuntimeError: torch.compile is not supported on Python 3.14+
|
||||
|
||||
Reference in New Issue
Block a user