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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   environs,
   5   fetchFromGitHub,
   6   gitpython,
   7   grpcio,
   8   grpcio-testing,
   9   minio,
  10   mmh3,
  11   pandas,
  12   pyarrow,
  13   pytestCheckHook,
  14   pythonOlder,
  15   requests,
  16   scikit-learn,
  17   setuptools-scm,
  18   ujson,
  19   wheel,
  20 }:
  21
  22 buildPythonPackage rec {
  23   pname = "pymilvus";
  24   version = "2.3.6";
  25   pyproject = true;
  26
  27   disabled = pythonOlder "3.7";
  28
  29   src = fetchFromGitHub {
  30     owner = "milvus-io";
  31     repo = "pymilvus";
  32     rev = "refs/tags/v${version}";
  33     hash = "sha256-K7k3MTOEm9+HDwMps9C8Al0Jmp1ptJw3pN1LEBOUz0U=";
  34   };
  35
  36   pythonRelaxDeps = [
  37     "environs"
  38     "grpcio"
  39   ];
  40
  41   nativeBuildInputs = [
  42     gitpython
  43     setuptools-scm
  44     wheel
  45   ];
  46
  47   propagatedBuildInputs = [
  48     environs
  49     grpcio
  50     minio
  51     mmh3
  52     pandas
  53     pyarrow
  54     requests
  55     ujson
  56   ];
  57
  58   nativeCheckInputs = [
  59     grpcio-testing
  60     pytestCheckHook
  61     scikit-learn
  62   ];
  63
  64   pythonImportsCheck = [ "pymilvus" ];
  65
  66   disabledTests = [ "test_get_commit" ];
  67
  68   meta = with lib; {
  69     description = "Python SDK for Milvus";
  70     homepage = "https://github.com/milvus-io/pymilvus";
  71     changelog = "https://github.com/milvus-io/pymilvus/releases/tag/v${version}";
  72     license = licenses.mit;
  73     maintainers = with maintainers; [ happysalada ];
  74   };
  75 }