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