diff --git a/pkgs/development/python-modules/torchtune/default.nix b/pkgs/development/python-modules/torchtune/default.nix index 61d4a5e1eefa..df6795d344f9 100644 --- a/pkgs/development/python-modules/torchtune/default.nix +++ b/pkgs/development/python-modules/torchtune/default.nix @@ -71,8 +71,7 @@ buildPythonPackage (finalAttrs: { # Not explicitly listed as requirements, but effectively imported at runtime torchao torchvision - ] - ++ huggingface-hub.optional-dependencies.hf_transfer; + ]; pythonImportsCheck = [ "torchtune" ];