diff options
Diffstat (limited to 'pkgs/development/python-modules/sagemaker/default.nix')
-rw-r--r-- | pkgs/development/python-modules/sagemaker/default.nix | 71 |
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 |