about summary refs log tree commit diff
path: root/pkgs/development/python-modules/sagemaker/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/sagemaker/default.nix')
-rw-r--r--pkgs/development/python-modules/sagemaker/default.nix71
1 files changed, 38 insertions, 33 deletions
diff --git a/pkgs/development/python-modules/sagemaker/default.nix b/pkgs/development/python-modules/sagemaker/default.nix
index 33e0c85697c1e..c96ad040dc2c5 100644
--- a/pkgs/development/python-modules/sagemaker/default.nix
+++ b/pkgs/development/python-modules/sagemaker/default.nix
@@ -1,38 +1,39 @@
-{ lib
-, buildPythonPackage
-, pythonOlder
-, fetchFromGitHub
-, fetchpatch
-, pythonRelaxDepsHook
-, setuptools
-, attrs
-, boto3
-, cloudpickle
-, google-pasta
-, numpy
-, protobuf
-, smdebug-rulesconfig
-, importlib-metadata
-, packaging
-, pandas
-, pathos
-, schema
-, pyyaml
-, jsonschema
-, platformdirs
-, tblib
-, urllib3
-, requests
-, docker
-, tqdm
-, psutil
-, scipy
-, accelerate
+{
+  lib,
+  buildPythonPackage,
+  pythonOlder,
+  fetchFromGitHub,
+  fetchpatch,
+  pythonRelaxDepsHook,
+  setuptools,
+  attrs,
+  boto3,
+  cloudpickle,
+  google-pasta,
+  numpy,
+  protobuf,
+  smdebug-rulesconfig,
+  importlib-metadata,
+  packaging,
+  pandas,
+  pathos,
+  schema,
+  pyyaml,
+  jsonschema,
+  platformdirs,
+  tblib,
+  urllib3,
+  requests,
+  docker,
+  tqdm,
+  psutil,
+  scipy,
+  accelerate,
 }:
 
 buildPythonPackage rec {
   pname = "sagemaker";
-  version = "2.217.0";
+  version = "2.219.0";
   pyproject = true;
 
   disabled = pythonOlder "3.8";
@@ -41,7 +42,7 @@ buildPythonPackage rec {
     owner = "aws";
     repo = "sagemaker-python-sdk";
     rev = "refs/tags/v${version}";
-    hash = "sha256-p73M0hBVpakdXGM56ECr4D3mQCBmB7Ud/p+Rj1+/Bic=";
+    hash = "sha256-TZpRRkoAlXU+Ccgxq49t+Cz0JOIUvYp7ok3x3sphncE=";
   };
 
   patches = [
@@ -99,7 +100,11 @@ buildPythonPackage rec {
   ];
 
   passthru.optional-dependencies = {
-    local = [ urllib3 docker pyyaml ];
+    local = [
+      urllib3
+      docker
+      pyyaml
+    ];
     scipy = [ scipy ];
     huggingface = [ accelerate ];
     # feature-processor = [ pyspark sagemaker-feature-store-pyspark ]; # not available in nixpkgs