pythonPackages.sagemaker: Init at 2.24.5

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 ];
+  };
+}

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 {