From e1d41aa50377449a6c521388d65f9ad8b4750e14 Mon Sep 17 00:00:00 2001
From: Austin Butler <austin@butler.lol>
Date: Thu, 31 Jul 2025 18:03:20 -0700
Subject: [PATCH 1/2] python3Packages.snapshot-restore-py: init at 1.0.0

---
 .../snapshot-restore-py/default.nix           | 36 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 ++
 2 files changed, 38 insertions(+)
 create mode 100644 pkgs/development/python-modules/snapshot-restore-py/default.nix

diff --git a/pkgs/development/python-modules/snapshot-restore-py/default.nix b/pkgs/development/python-modules/snapshot-restore-py/default.nix
new file mode 100644
index 000000000000..bb478bc8d11c
--- /dev/null
+++ b/pkgs/development/python-modules/snapshot-restore-py/default.nix
@@ -0,0 +1,36 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+  isPy27,
+  pytestCheckHook,
+  setuptools,
+}:
+buildPythonPackage rec {
+  pname = "snapshot-restore-py";
+  version = "1.0.0";
+  pyproject = true;
+
+  disabled = isPy27;
+
+  src = fetchFromGitHub {
+    owner = "aws";
+    repo = "snapshot-restore-py";
+    tag = "v${version}";
+    hash = "sha256-sixVSQcEqLTUrKxYAM13gzqttWnbXPMII0V/gtXM1IE=";
+  };
+
+  build-system = [ setuptools ];
+
+  nativeCheckInputs = [ pytestCheckHook ];
+
+  pythonImportsCheck = [ "snapshot_restore_py" ];
+
+  meta = {
+    description = "Snapshot Restore for Python library which can be used for registering runtime hooks in Snapstart enabled Python Lambda functions";
+    homepage = "https://github.com/aws/snapshot-restore-py";
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ austinbutler ];
+    platforms = lib.platforms.all;
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index b2110427df73..d6f0da38edec 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -16687,6 +16687,8 @@ self: super: with self; {
 
   snappy-manifolds = callPackage ../development/python-modules/snappy-manifolds { };
 
+  snapshot-restore-py = callPackage ../development/python-modules/snapshot-restore-py { };
+
   snapshottest = callPackage ../development/python-modules/snapshottest { };
 
   snaptime = callPackage ../development/python-modules/snaptime { };

From 43dd9f2593e6269db4607f0615edb604e10fda47 Mon Sep 17 00:00:00 2001
From: Austin Butler <austin@butler.lol>
Date: Thu, 31 Jul 2025 18:03:41 -0700
Subject: [PATCH 2/2] python3Packages.awslambdaric: 3.0.0 -> 3.1.1

---
 .../python-modules/awslambdaric/default.nix            | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/awslambdaric/default.nix b/pkgs/development/python-modules/awslambdaric/default.nix
index 99104b8b2724..fd34cef6ec6b 100644
--- a/pkgs/development/python-modules/awslambdaric/default.nix
+++ b/pkgs/development/python-modules/awslambdaric/default.nix
@@ -1,5 +1,4 @@
 {
-  stdenv,
   lib,
   buildPythonPackage,
   fetchFromGitHub,
@@ -14,10 +13,11 @@
   perl,
   setuptools,
   simplejson,
+  snapshot-restore-py,
 }:
 buildPythonPackage rec {
   pname = "awslambdaric";
-  version = "3.0.0";
+  version = "3.1.1";
   pyproject = true;
 
   disabled = isPy27;
@@ -29,7 +29,10 @@ buildPythonPackage rec {
     sha256 = "sha256-pUVWd4zpmTygndPIy76uVk7+sLCmwQqulLaUI7B0fQc=";
   };
 
-  propagatedBuildInputs = [ simplejson ];
+  propagatedBuildInputs = [
+    simplejson
+    snapshot-restore-py
+  ];
 
   nativeBuildInputs = [
     autoconf271
@@ -55,7 +58,6 @@ buildPythonPackage rec {
   ];
 
   meta = with lib; {
-    broken = stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64;
     description = "AWS Lambda Runtime Interface Client for Python";
     homepage = "https://github.com/aws/aws-lambda-python-runtime-interface-client";
     license = licenses.asl20;