From bb4a735053e973f274452f560bef42ba106ae627 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Robert=20Sch=C3=BCtz?= <nix@dotlambda.de>
Date: Thu, 15 Dec 2022 11:23:37 -0800
Subject: [PATCH] python310Packages.elegy: use poetry-core

---
 pkgs/development/python-modules/elegy/default.nix | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/elegy/default.nix b/pkgs/development/python-modules/elegy/default.nix
index d4df308e679c..290442d5152a 100644
--- a/pkgs/development/python-modules/elegy/default.nix
+++ b/pkgs/development/python-modules/elegy/default.nix
@@ -5,8 +5,9 @@
 , deepmerge
 , dm-haiku
 , fetchFromGitHub
+, fetchpatch
 , jaxlib
-, poetry
+, poetry-core
 , pytestCheckHook
 , pythonOlder
 , pyyaml
@@ -35,6 +36,14 @@ buildPythonPackage rec {
     hash = "sha256-FZmLriYhsX+zyQKCtCjbOy6MH+AvjzHRNUyaDSXGlLI=";
   };
 
+  patches = [
+    (fetchpatch {
+      name = "use-poetry-core.patch";
+      url = "https://github.com/poets-ai/elegy/commit/0ed472882f470ed9eb7a63b8a537ffabe7e19aa7.patch";
+      hash = "sha256-nO/imHo7tEsiZh+64CF/M4eXQ1so3IunVhv8CvYP1ks=";
+    })
+  ];
+
   # The cloudpickle constraint is too strict. wandb is marked as an optional
   # dependency but `buildPythonPackage` doesn't seem to respect that setting.
   # Python constraint: https://github.com/poets-ai/elegy/issues/244
@@ -46,7 +55,7 @@ buildPythonPackage rec {
   '';
 
   nativeBuildInputs = [
-    poetry
+    poetry-core
   ];
 
   buildInputs = [