Commit 59adbcdf authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent b8ec1309
Loading
Loading
Loading
Loading
+5 −16
Original line number Diff line number Diff line
@@ -2,7 +2,6 @@
  lib,
  buildPythonPackage,
  fetchFromGitHub,
  fetchpatch,

  # build-system
  hatchling,
@@ -25,6 +24,7 @@
  pyyaml,
  requests,
  sagemaker-core,
  sagemaker-mlflow,
  schema,
  smdebug-rulesconfig,
  tblib,
@@ -38,29 +38,16 @@

buildPythonPackage rec {
  pname = "sagemaker";
  version = "2.232.1";
  version = "2.232.2";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "aws";
    repo = "sagemaker-python-sdk";
    rev = "refs/tags/v${version}";
    hash = "sha256-I+iZKx1CnZIGYgYuYhhs8BnY84KPyKOGw8M0He26DGU=";
    hash = "sha256-q0JxQi1kUnp5L/hexxpGhR4t0v53l8iVuc9H6N0K6Y4=";
  };

  patches = [
    # Distutils removal, fix build with python 3.12
    # https://github.com/aws/sagemaker-python-sdk/pull/4544
    (fetchpatch {
      url = "https://github.com/aws/sagemaker-python-sdk/commit/84447ba59e544c810aeb842fd058e20d89e3fc74.patch";
      hash = "sha256-B8Q18ViB7xYy1F5LoL1NvXj2lnFPgt+C9wssSODyAXM=";
    })
    (fetchpatch {
      url = "https://github.com/aws/sagemaker-python-sdk/commit/e9e08a30cb42d4b2d7299c1c4b42d680a8c78110.patch";
      hash = "sha256-uGPtXSXfeaIvt9kkZZKQDuiZfoRgw3teffuxai1kKlY=";
    })
  ];

  build-system = [
    hatchling
  ];
@@ -69,6 +56,7 @@ buildPythonPackage rec {
    "boto3"
    "cloudpickle"
    "importlib-metadata"
    "protobuf"
  ];

  dependencies = [
@@ -89,6 +77,7 @@ buildPythonPackage rec {
    pyyaml
    requests
    sagemaker-core
    sagemaker-mlflow
    schema
    smdebug-rulesconfig
    tblib