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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , attrs
   5 , boto3
   6 , google-pasta
   7 , importlib-metadata
   8 , numpy
   9 , protobuf
  10 , protobuf3-to-dict
  11 , smdebug-rulesconfig
  12 , pandas
  13 , packaging
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "sagemaker";
  18   version = "2.35.0";
  19
  20   src = fetchPypi {
  21     inherit pname version;
  22     sha256 = "sha256-12YYUZbctM6oRaC7Sr/hOghAM+s/Cdm5XWHaVU5Gg6Q=";
  23   };
  24
  25   pythonImportsCheck = [
  26     "sagemaker"
  27     "sagemaker.lineage.visualizer"
  28   ];
  29
  30   propagatedBuildInputs = [
  31     attrs
  32     boto3
  33     google-pasta
  34     importlib-metadata
  35     numpy
  36     packaging
  37     protobuf
  38     protobuf3-to-dict
  39     smdebug-rulesconfig
  40     pandas
  41   ];
  42
  43   doCheck = false;
  44
  45   postFixup = ''
  46     [ "$($out/bin/sagemaker-upgrade-v2 --help 2>&1 | grep -cim1 'pandas failed to import')" -eq "0" ]
  47   '';
  48
  49   meta = with lib; {
  50     description = "Library for training and deploying machine learning models on Amazon SageMaker";
  51     homepage = "https://github.com/aws/sagemaker-python-sdk/";
  52     license = licenses.asl20;
  53     maintainers = with maintainers; [ nequissimus ];
  54   };
  55 }