diff --git a/pkgs/development/python-modules/sagemaker/default.nix b/pkgs/development/python-modules/sagemaker/default.nix new file mode 100644 index 000000000000..048e7e1808c6 --- /dev/null +++ b/pkgs/development/python-modules/sagemaker/default.nix @@ -0,0 +1,35 @@ +{ lib, buildPythonPackage, fetchPypi, attrs, boto3, google-pasta +, importlib-metadata, numpy, protobuf, protobuf3-to-dict, smdebug-rulesconfig }: + +buildPythonPackage rec { + pname = "sagemaker"; + version = "2.24.5"; + + src = fetchPypi { + inherit pname version; + sha256 = "1j1a058ic00yxnf0cc364fzn82pacih5ffrh5s4dw1q4s3by4cvd"; + }; + + doCheck = false; + + pythonImportsCheck = [ "sagemaker" ]; + + propagatedBuildInputs = [ + attrs + boto3 + google-pasta + importlib-metadata + numpy + protobuf + protobuf3-to-dict + smdebug-rulesconfig + ]; + + meta = with lib; { + description = + "Library for training and deploying machine learning models on Amazon SageMaker"; + homepage = "https://github.com/aws/sagemaker-python-sdk/"; + license = licenses.asl20; + maintainers = with maintainers; [ nequissimus ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 6b6861db3b0c..cd687874f79c 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -7020,6 +7020,8 @@ in { safety = callPackage ../development/python-modules/safety { }; + sagemaker = callPackage ../development/python-modules/sagemaker { }; + salmon-mail = callPackage ../development/python-modules/salmon-mail { }; sane = callPackage ../development/python-modules/sane {