From fec04ec93bf2d4dd83e55dd45936dc7e9321867a Mon Sep 17 00:00:00 2001 From: Madoura Date: Tue, 24 Oct 2023 13:55:43 -0500 Subject: [PATCH] python3Packages.torchinfo: disable failing test --- pkgs/development/python-modules/torchinfo/default.nix | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pkgs/development/python-modules/torchinfo/default.nix b/pkgs/development/python-modules/torchinfo/default.nix index 6cc7bcf3e27c..59b7e99c9dfc 100644 --- a/pkgs/development/python-modules/torchinfo/default.nix +++ b/pkgs/development/python-modules/torchinfo/default.nix @@ -53,6 +53,8 @@ buildPythonPackage rec { "test_flan_t5_small" # AssertionError in output "test_google" + # "addmm_impl_cpu_" not implemented for 'Half' + "test_input_size_half_precision" ]; disabledTestPaths = [