Unverified Commit 6ccfebec authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.xgrammar: update dependencies (#455955)

parents 6b0d2f6d ceaf71d9

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

+6 −4
Original line number Diff line number Diff line
@@ -12,15 +12,16 @@ 


  # dependencies

  mlx-lm,

  numpy,

  pydantic,

  sentencepiece,

  tiktoken,

  torch,

  transformers,

  triton,



  # tests

  pytestCheckHook,

  sentencepiece,

  tiktoken,

  writableTmpDirAsHomeHook,

}:
@@ -50,9 +51,8 @@ buildPythonPackage rec { 
  dontUseCmakeConfigure = true;



  dependencies = [

    numpy

    pydantic

    sentencepiece

    tiktoken

    torch

    transformers

  ]
@@ -65,6 +65,8 @@ buildPythonPackage rec { 


  nativeCheckInputs = [

    pytestCheckHook

    sentencepiece

    tiktoken

    writableTmpDirAsHomeHook

  ];