python312Packages.sagemaker: 2.224.1 -> 2.232.1

Diff: aws/sagemaker-python-sdk@refs/tags/v2.224.1...v2.232.1 Changelog: https://github.com/aws/sagemaker-python-sdk/blob/v2.232.1/CHANGELOG.md
NixOS · Oct 3, 2024 · 32b2c9d · 32b2c9d
1 parent 83bec9a
commit 32b2c9d
Showing 1 changed file with 36 additions and 30 deletions.
diff --git a/pkgs/development/python-modules/sagemaker/default.nix b/pkgs/development/python-modules/sagemaker/default.nix
@@ -1,47 +1,51 @@
 {
   lib,
   buildPythonPackage,
-  pythonOlder,
   fetchFromGitHub,
   fetchpatch,
-  setuptools,
+
+  # build-system
+  hatchling,
+
+  # dependencies
   attrs,
   boto3,
   cloudpickle,
+  docker,
   google-pasta,
-  numpy,
-  protobuf,
-  smdebug-rulesconfig,
   importlib-metadata,
+  jsonschema,
+  numpy,
   packaging,
   pandas,
   pathos,
-  schema,
-  pyyaml,
-  jsonschema,
   platformdirs,
-  tblib,
-  urllib3,
+  protobuf,
+  psutil,
+  pyyaml,
   requests,
-  docker,
+  sagemaker-core,
+  schema,
+  smdebug-rulesconfig,
+  tblib,
   tqdm,
-  psutil,
+  urllib3,
+
+  # optional-dependencies
   scipy,
   accelerate,
 }:
 
 buildPythonPackage rec {
   pname = "sagemaker";
-  version = "2.224.1";
+  version = "2.232.1";
   pyproject = true;
 
-  disabled = pythonOlder "3.8";
-
   src = fetchFromGitHub {
     owner = "aws";
     repo = "sagemaker-python-sdk";
     rev = "refs/tags/v${version}";
-    hash = "sha256-Kc66sygHGFqMvSY7rACb62wJEJesnN4KDmtYZLIOsqc=";
+    hash = "sha256-I+iZKx1CnZIGYgYuYhhs8BnY84KPyKOGw8M0He26DGU=";
   };
 
   patches = [
@@ -58,10 +62,11 @@ buildPythonPackage rec {
   ];
 
   build-system = [
-    setuptools
+    hatchling
   ];
 
   pythonRelaxDeps = [
+    "boto3"
     "cloudpickle"
     "importlib-metadata"
   ];
@@ -70,24 +75,25 @@ buildPythonPackage rec {
     attrs
     boto3
     cloudpickle
+    docker
     google-pasta
-    numpy
-    protobuf
-    smdebug-rulesconfig
     importlib-metadata
+    jsonschema
+    numpy
     packaging
     pandas
     pathos
-    schema
-    pyyaml
-    jsonschema
     platformdirs
-    tblib
-    urllib3
+    protobuf
+    psutil
+    pyyaml
     requests
-    docker
+    sagemaker-core
+    schema
+    smdebug-rulesconfig
+    tblib
     tqdm
-    psutil
+    urllib3
   ];
 
   doCheck = false; # many test dependencies are not available in nixpkgs
@@ -108,11 +114,11 @@ buildPythonPackage rec {
     # feature-processor = [ pyspark sagemaker-feature-store-pyspark ]; # not available in nixpkgs
   };
 
-  meta = with lib; {
+  meta = {
     description = "Library for training and deploying machine learning models on Amazon SageMaker";
     homepage = "https://github.com/aws/sagemaker-python-sdk/";
     changelog = "https://github.com/aws/sagemaker-python-sdk/blob/v${version}/CHANGELOG.md";
-    license = licenses.asl20;
-    maintainers = with maintainers; [ nequissimus ];
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ nequissimus ];
   };
 }