diff --git a/pkgs/development/python-modules/torchcodec/default.nix b/pkgs/development/python-modules/torchcodec/default.nix index 22cc9d90482e..5bca23468383 100644 --- a/pkgs/development/python-modules/torchcodec/default.nix +++ b/pkgs/development/python-modules/torchcodec/default.nix @@ -156,72 +156,75 @@ buildPythonPackage.override { inherit (torch) stdenv; } (finalAttrs: { "test/plugin/test_plugins.py" ]; - disabledTests = - lib.optionals rocmSupport [ - # HSA runtime logs topology error in sandbox breaking test that asserts no output - "test_python_logger" - ] - ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ - # Fails in the sandbox: - # Error in cpuinfo: failed to parse the list of possible processors in /sys/devices/system/cpu/possible - "test_python_logger" + disabledTests = [ + # AssertionError: Tensor-likes are not close! + "test_audio_against_cli" + ] + ++ lib.optionals rocmSupport [ + # HSA runtime logs topology error in sandbox breaking test that asserts no output + "test_python_logger" + ] + ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ + # Fails in the sandbox: + # Error in cpuinfo: failed to parse the list of possible processors in /sys/devices/system/cpu/possible + "test_python_logger" - # AssertionError: index 0 - "test_get_frames_played_at" + # AssertionError: index 0 + "test_get_frames_played_at" - # AssertionError: Tensor-likes are not equal! - "test_against_cli" - "test_against_ref" - "test_color_conversion_library" - "test_color_conversion_library_with_dimension_order" - "test_compile" - "test_compile_seek_and_next" - "test_create_decoder" - "test_crop_transform" - "test_custom_frame_mappings_json_and_bytes" - "test_file_like_decoding" - "test_get_frame_at" - "test_get_frame_at_av1" - "test_get_frame_at_index" - "test_get_frame_at_pts" - "test_get_frame_played_at" - "test_get_frame_played_at_h265" - "test_get_frame_with_info_at_index" - "test_get_frames_at" - "test_get_frames_at_indices" - "test_get_frames_at_indices_negative_indices" - "test_get_frames_by_pts_in_range" - "test_get_frames_in_range" - "test_get_frames_in_range_slice_indices_syntax" - "test_get_frames_with_missing_num_frames_metadata" - "test_getitem_int" - "test_getitem_numpy_int" - "test_getitem_slice" - "test_iteration" - "test_seek_and_next" - "test_seek_mode_custom_frame_mappings" - "test_seek_to_negative_pts" - "test_throws_exception_at_eof" - ] - ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [ - # RuntimeError: Invalid AVIO context holder - "test_1d_samples" - "test_against_cli" - "test_against_to_file" - "test_against_to_file" - "test_contiguit" - "test_crf_valid_value" - "test_encode_to_tensor_long_outpu" - "test_num_channels" - "test_round_trip" - "test_video_encoder_against_ffmpeg_cli" - "test_video_encoder_round_trip" + # AssertionError: Tensor-likes are not equal! + "test_against_cli" + "test_against_ref" + "test_color_conversion_library" + "test_color_conversion_library_with_dimension_order" + "test_compile" + "test_compile_seek_and_next" + "test_create_decoder" + "test_crop_transform" + "test_custom_frame_mappings_json_and_bytes" + "test_file_like_decoding" + "test_get_frame_at" + "test_get_frame_at_av1" + "test_get_frame_at_index" + "test_get_frame_at_pts" + "test_get_frame_played_at" + "test_get_frame_played_at_h265" + "test_get_frame_with_info_at_index" + "test_get_frames_at" + "test_get_frames_at_indices" + "test_get_frames_at_indices_negative_indices" + "test_get_frames_by_pts_in_range" + "test_get_frames_in_range" + "test_get_frames_in_range_slice_indices_syntax" + "test_get_frames_with_missing_num_frames_metadata" + "test_getitem_int" + "test_getitem_numpy_int" + "test_getitem_slice" + "test_iteration" + "test_seek_and_next" + "test_seek_mode_custom_frame_mappings" + "test_seek_to_negative_pts" + "test_throws_exception_at_eof" + ] + ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [ + # RuntimeError: Invalid AVIO context holder + "test_1d_samples" + "test_against_cli" + "test_against_to_file" + "test_against_to_file" + "test_contiguit" + "test_crf_valid_value" + "test_encode_to_tensor_long_outpu" + "test_num_channels" + "test_round_trip" + "test_video_encoder_against_ffmpeg_cli" + "test_video_encoder_round_trip" - # RuntimeError: Requested next frame while there are no more frames left to decode - "test_next" - "test_throws_exception_at_eof" - "test_throws_exception_if_seek_too_far" - ]; + # RuntimeError: Requested next frame while there are no more frames left to decode + "test_next" + "test_throws_exception_at_eof" + "test_throws_exception_if_seek_too_far" + ]; meta = { description = "PyTorch media decoding and encoding";