cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

pythonPackages.sagemaker: add pandas for analytical feature

Browse source
This commit is contained in:
Sandro Jäckel 2021-02-22 08:39:00 +01:00
parent 9ab61d974e
commit e677462813
No known key found for this signature in database
GPG key ID: 3AF5A43A3EECC2E5
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -9,6 +9,7 @@
, protobuf , protobuf
, protobuf3-to-dict , protobuf3-to-dict
, smdebug-rulesconfig , smdebug-rulesconfig
, pandas
}: }:
buildPythonPackage rec { buildPythonPackage rec {
@ -20,9 +21,10 @@ buildPythonPackage rec {
sha256 = "sha256-xQ1nt8FcjuoilzM5PbU8KHgirPyj9us+ykyjfgEqZhg="; sha256 = "sha256-xQ1nt8FcjuoilzM5PbU8KHgirPyj9us+ykyjfgEqZhg=";
}; };
doCheck = false; pythonImportsCheck = [
"sagemaker"
pythonImportsCheck = [ "sagemaker" ]; "sagemaker.lineage.visualizer"
];
propagatedBuildInputs = [ propagatedBuildInputs = [
attrs attrs
@ -33,8 +35,15 @@ buildPythonPackage rec {
protobuf protobuf
protobuf3-to-dict protobuf3-to-dict
smdebug-rulesconfig smdebug-rulesconfig
pandas
]; ];
doCheck = false;
postFixup = ''
[ "$($out/bin/sagemaker-upgrade-v2 --help 2>&1 | grep -cim1 'pandas failed to import')" -eq "0" ]
'';
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/";