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

python3Packages.mistral-common: exclude optional sentencepiece dependency on Darwin (#466361)

parents 600c3b0c fd48e390

pkgs/development/python-modules/mistral-common/default.nix

+20 −4
Original line number Diff line number Diff line 
{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,
@@ -14,7 +15,6 @@ 
  pydantic,

  pydantic-extra-types,

  requests,

  sentencepiece,

  tiktoken,

  typing-extensions,
@@ -26,6 +26,7 @@ 
  pycountry,

  pydantic-settings,

  pytestCheckHook,

  sentencepiece,

  soundfile,

  soxr,

  uvicorn,
@@ -53,7 +54,6 @@ buildPythonPackage rec { 
    pydantic

    pydantic-extra-types

    requests

    sentencepiece

    tiktoken

    typing-extensions

  ];
@@ -62,7 +62,8 @@ buildPythonPackage rec { 
    opencv = [

      opencv-python-headless

    ];

    sentencepiece = [

    # Broken on Darwin. See https://github.com/NixOS/nixpkgs/issues/466092

    sentencepiece = lib.optionals (!stdenv.hostPlatform.isDarwin) [

      sentencepiece

    ];

    soundfile = [
@@ -97,7 +98,8 @@ buildPythonPackage rec { 
    soundfile

    soxr

    uvicorn

  ];

  ]

  ++ lib.concatAttrValues optional-dependencies;



  disabledTests = [

    # Require internet
@@ -112,6 +114,20 @@ buildPythonPackage rec { 
    "test_openai_chat_fields"

  ];



  # Requires sentencepiece which segfaults when initialized on Darwin

  # See https://github.com/NixOS/nixpkgs/issues/466092

  disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [

    "tests/experimental/test_app.py"

    "tests/experimental/test_tools.py"

    "tests/test_fim_tokenizer.py"

    "tests/test_integration_samples.py"

    "tests/test_mistral_tokenizer.py"

    "tests/test_tokenize_v1.py"

    "tests/test_tokenize_v2.py"

    "tests/test_tokenize_v3.py"

    "tests/test_tokenizer_v7.py"

  ];



  meta = {

    description = "Tools to help you work with Mistral models";

    homepage = "https://github.com/mistralai/mistral-common";