python3Packages.sagemaker: 2.135.0 -> 2.173.0
This commit is contained in:
parent
2db6614523
commit
686484946d
|
@ -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/";
|
||||||
|
|
Loading…
Reference in a new issue