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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5
   6   # build-system
   7   hatchling,
   8
   9   # dependencies
  10   attrs,
  11   boto3,
  12   cloudpickle,
  13   docker,
  14   fastapi,
  15   google-pasta,
  16   importlib-metadata,
  17   jsonschema,
  18   numpy,
  19   omegaconf,
  20   packaging,
  21   pandas,
  22   pathos,
  23   platformdirs,
  24   protobuf,
  25   psutil,
  26   pyyaml,
  27   requests,
  28   sagemaker-core,
  29   schema,
  30   smdebug-rulesconfig,
  31   tblib,
  32   tqdm,
  33   urllib3,
  34   uvicorn,
  35
  36   # optional-dependencies
  37   scipy,
  38   accelerate,
  39 }:
  40
  41 buildPythonPackage rec {
  42   pname = "sagemaker";
  43   version = "2.239.0";
  44   pyproject = true;
  45
  46   src = fetchFromGitHub {
  47     owner = "aws";
  48     repo = "sagemaker-python-sdk";
  49     tag = "v${version}";
  50     hash = "sha256-KE3EWCxl4Q90pChswYjmBwtPazAjVw2MaAjNesQoWiM=";
  51   };
  52
  53   build-system = [
  54     hatchling
  55   ];
  56
  57   pythonRelaxDeps = [
  58     "attrs"
  59     "boto3"
  60     "cloudpickle"
  61     "importlib-metadata"
  62     "numpy"
  63     "omegaconf"
  64     "protobuf"
  65   ];
  66
  67   dependencies = [
  68     attrs
  69     boto3
  70     cloudpickle
  71     docker
  72     fastapi
  73     google-pasta
  74     importlib-metadata
  75     jsonschema
  76     numpy
  77     omegaconf
  78     packaging
  79     pandas
  80     pathos
  81     platformdirs
  82     protobuf
  83     psutil
  84     pyyaml
  85     requests
  86     sagemaker-core
  87     schema
  88     smdebug-rulesconfig
  89     tblib
  90     tqdm
  91     urllib3
  92     uvicorn
  93   ];
  94
  95   doCheck = false; # many test dependencies are not available in nixpkgs
  96
  97   pythonImportsCheck = [
  98     "sagemaker"
  99     "sagemaker.lineage.visualizer"
 100   ];
 101
 102   optional-dependencies = {
 103     local = [
 104       urllib3
 105       docker
 106       pyyaml
 107     ];
 108     scipy = [ scipy ];
 109     huggingface = [ accelerate ];
 110     # feature-processor = [ pyspark sagemaker-feature-store-pyspark ]; # not available in nixpkgs
 111   };
 112
 113   meta = {
 114     description = "Library for training and deploying machine learning models on Amazon SageMaker";
 115     homepage = "https://github.com/aws/sagemaker-python-sdk/";
 116     changelog = "https://github.com/aws/sagemaker-python-sdk/blob/v${version}/CHANGELOG.md";
 117     license = lib.licenses.asl20;
 118     maintainers = with lib.maintainers; [ nequissimus ];
 119   };
 120 }