Commit 0d415c83 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.huggingface-hub: add optional-dependencies

parent 559834db

pkgs/development/python-modules/huggingface-hub/default.nix

+57 −0
Original line number Diff line number Diff line
@@ -14,6 +14,27 @@ 
  requests,

  tqdm,

  typing-extensions,



  # optional-dependencies

  # cli

  inquirerpy,

  # inference

  aiohttp,

  # torch

  torch,

  safetensors,

  # hf_transfer

  hf-transfer,

  # fastai

  toml,

  fastai,

  fastcore,

  # tensorflow

  tensorflow,

  pydot,

  graphviz,

  # tensorflow-testing

  keras,

}:



buildPythonPackage rec {
@@ -40,6 +61,42 @@ buildPythonPackage rec { 
    typing-extensions

  ];



  optional-dependencies = {

    all = [



    ];

    cli = [

      inquirerpy

    ];

    inference = [

      aiohttp

    ];

    torch = [

      torch

      safetensors

    ] ++ safetensors.optional-dependencies.torch;

    hf_transfer = [

      hf-transfer

    ];

    fastai = [

      toml

      fastai

      fastcore

    ];

    tensorflow = [

      tensorflow

      pydot

      graphviz

    ];

    tensorflow-testing = [

      tensorflow

      keras

    ];

    hf_xet = [

      # hf-xet

    ];

  };



  # Tests require network access.

  doCheck = false;