b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

python3Packages.sagemaker: 2.135.0 -> 2.173.0

Browse source
This commit is contained in:
Nick Cao 2023-07-30 10:59:01 +08:00
parent 2db6614523
commit 686484946d
No known key found for this signature in database
1 changed files with 41 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
pkgs/development/python-modules/sagemaker/default.nix
View file

@ -1,65 +1,84 @@
{ lib { lib
, buildPythonPackage , buildPythonPackage
, fetchPypi , pythonOlder
, fetchFromGitHub
, pythonRelaxDepsHook , pythonRelaxDepsHook
, attrs , attrs
, boto3 , boto3
, cloudpickle
, google-pasta , google-pasta
, importlib-metadata
, numpy , numpy
, protobuf , protobuf
, protobuf3-to-dict
, smdebug-rulesconfig , smdebug-rulesconfig
, importlib-metadata
, packaging
, pandas , pandas
, pathos , pathos
, packaging , schema
, pythonOlder , pyyaml
, jsonschema
, platformdirs
, tblib
, urllib3
, docker
, scipy
}: }:
buildPythonPackage rec { buildPythonPackage rec {
pname = "sagemaker"; pname = "sagemaker";
version = "2.135.0"; version = "2.173.0";
format = "setuptools"; format = "setuptools";
disabled = pythonOlder "3.7"; disabled = pythonOlder "3.6";
src = fetchPypi { src = fetchFromGitHub {
inherit pname version; owner = "aws";
hash = "sha256-ypdcqEYLxHbfnq1ycq3hVLThhIIs3pq29Fv33Ly2hbE="; repo = "sagemaker-python-sdk";
rev = "refs/tags/v${version}";
hash = "sha256-Fzkw37bRBbCD7VBIsN4Qkk6dI0Qh4Gvt5TJbnoUuPCs=";
}; };
nativeBuildInputs = [ pythonRelaxDepsHook ]; nativeBuildInputs = [
pythonRelaxDepsHook
];
pythonRelaxDeps = [ pythonRelaxDeps = [
# FIXME: Remove when >= 2.111.0
"attrs" "attrs"
"protobuf" "boto3"
]; ];
propagatedBuildInputs = [ propagatedBuildInputs = [
attrs attrs
boto3 boto3
cloudpickle
google-pasta google-pasta
importlib-metadata
numpy numpy
packaging
pathos
protobuf protobuf
protobuf3-to-dict
smdebug-rulesconfig smdebug-rulesconfig
importlib-metadata
packaging
pandas pandas
pathos
schema
pyyaml
jsonschema
platformdirs
tblib
]; ];
postFixup = '' doCheck = false; # many test dependencies are not available in nixpkgs
[ "$($out/bin/sagemaker-upgrade-v2 --help 2>&1 | grep -cim1 'pandas failed to import')" -eq "0" ]
'';
doCheck = false;
pythonImportsCheck = [ pythonImportsCheck = [
"sagemaker" "sagemaker"
"sagemaker.lineage.visualizer" "sagemaker.lineage.visualizer"
]; ];
passthru.optional-dependencies = {
local = [ urllib3 docker pyyaml ];
scipy = [ scipy ];
# feature-processor = [ pyspark sagemaker-feature-store-pyspark ]; # not available in nixpkgs
};
meta = with lib; { meta = with lib; {
description = "Library for training and deploying machine learning models on Amazon SageMaker"; description = "Library for training and deploying machine learning models on Amazon SageMaker";
homepage = "https://github.com/aws/sagemaker-python-sdk/"; homepage = "https://github.com/aws/sagemaker-python-sdk/";