pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, attrs
5, boto3
6, google-pasta
7, importlib-metadata
8, numpy
9, protobuf
10, protobuf3-to-dict
11, smdebug-rulesconfig
12, pandas
13, pathos
14, packaging
15}:
16
17buildPythonPackage rec {
18 pname = "sagemaker";
19 version = "2.63.1";
20
21 src = fetchPypi {
22 inherit pname version;
23 sha256 = "924847e9793b76d188049718aabbcad975296bb267812ad18e0279e7af0cb748";
24 };
25
26 pythonImportsCheck = [
27 "sagemaker"
28 "sagemaker.lineage.visualizer"
29 ];
30
31 propagatedBuildInputs = [
32 attrs
33 boto3
34 google-pasta
35 importlib-metadata
36 numpy
37 packaging
38 pathos
39 protobuf
40 protobuf3-to-dict
41 smdebug-rulesconfig
42 pandas
43 ];
44
45 doCheck = false;
46
47 postFixup = ''
48 [ "$($out/bin/sagemaker-upgrade-v2 --help 2>&1 | grep -cim1 'pandas failed to import')" -eq "0" ]
49 '';
50
51 meta = with lib; {
52 description = "Library for training and deploying machine learning models on Amazon SageMaker";
53 homepage = "https://github.com/aws/sagemaker-python-sdk/";
54 license = licenses.asl20;
55 maintainers = with maintainers; [ nequissimus ];
56 };
57}