diff --git a/pkgs/development/python-modules/safetensors/default.nix b/pkgs/development/python-modules/safetensors/default.nix index b0a20ac9e064..7a695ce0d610 100644 --- a/pkgs/development/python-modules/safetensors/default.nix +++ b/pkgs/development/python-modules/safetensors/default.nix @@ -19,6 +19,7 @@ pytest, pytest-benchmark, hypothesis, + fsspec, # tests pytestCheckHook, @@ -81,6 +82,7 @@ buildPythonPackage rec { pytest pytest-benchmark hypothesis + fsspec ]; all = self.torch ++ self.numpy ++ self.pinned-tf ++ self.jax ++ self.paddlepaddle ++ self.testing; dev = self.all; @@ -91,6 +93,7 @@ buildPythonPackage rec { numpy pytestCheckHook torch + fsspec ]; enabledTestPaths = [ "tests" ];