tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
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, pythonOlder
16}:
17
18buildPythonPackage rec {
19 pname = "sagemaker";
20 version = "2.91.1";
21 format = "setuptools";
22
23 disabled = pythonOlder "3.7";
24
25 src = fetchPypi {
26 inherit pname version;
27 hash = "sha256-DP6bdakmOX2qiqbkz/D4Q/aexjy7dTDXlDKvnCf3SaA=";
28 };
29
30 propagatedBuildInputs = [
31 attrs
32 boto3
33 google-pasta
34 importlib-metadata
35 numpy
36 packaging
37 pathos
38 protobuf
39 protobuf3-to-dict
40 smdebug-rulesconfig
41 pandas
42 ];
43
44 postPatch = ''
45 substituteInPlace setup.py \
46 --replace "attrs==20.3.0" "attrs>=20.3.0"
47 '';
48
49 postFixup = ''
50 [ "$($out/bin/sagemaker-upgrade-v2 --help 2>&1 | grep -cim1 'pandas failed to import')" -eq "0" ]
51 '';
52
53 doCheck = false;
54
55 pythonImportsCheck = [
56 "sagemaker"
57 "sagemaker.lineage.visualizer"
58 ];
59
60 meta = with lib; {
61 description = "Library for training and deploying machine learning models on Amazon SageMaker";
62 homepage = "https://github.com/aws/sagemaker-python-sdk/";
63 license = licenses.asl20;
64 maintainers = with maintainers; [ nequissimus ];
65 };
66}