python310Packages.elegy: use poetry-core

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 = [