Unverified Commit 2af57377 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.safetensors: add fsspec (#441390)

parents b3e206ec 443fdc03

pkgs/development/python-modules/safetensors/default.nix

+3 −0
Original line number Diff line number Diff line
@@ -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" ];