python3Packages.mmengine: fix pytorch 2.5 compatibility

2024-11-02 12:34:22 +01:00
parent 5c7d5fbce4
commit 3ebbc8add8
1 changed files with 9 additions and 0 deletions
@@ -3,6 +3,7 @@
  stdenv,
  buildPythonPackage,
  fetchFromGitHub,
+  fetchpatch2,

  # build-system
  setuptools,
@@ -41,6 +42,14 @@ buildPythonPackage rec {
    hash = "sha256-bZ6O4UOYUCwq11YmgRWepOIngYxYD/fNfM/VmcyUv9k=";
  };

+  patches = [
+    (fetchpatch2 {
+      name = "mmengine-torch-2.5-compat.patch";
+      url = "https://github.com/open-mmlab/mmengine/commit/4c22f78cdea2981a2b48a167e9feffe4721f8901.patch";
+      hash = "sha256-k+IFLeqTEVUGGiqmZg56LK64H/UTvpGN20GJT59wf4A=";
+    })
+  ];
+
  build-system = [ setuptools ];

  dependencies = [