From 74a7d53e6898871876f46ed912095c2557bb210a Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Sat, 13 Jan 2024 13:31:52 +0100
Subject: [PATCH] python311Packages.rotary-embedding-torch: add beartype

---
 .../python-modules/rotary-embedding-torch/default.nix          | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkgs/development/python-modules/rotary-embedding-torch/default.nix b/pkgs/development/python-modules/rotary-embedding-torch/default.nix
index 91bb5a00231c..f9467f63e48c 100644
--- a/pkgs/development/python-modules/rotary-embedding-torch/default.nix
+++ b/pkgs/development/python-modules/rotary-embedding-torch/default.nix
@@ -7,6 +7,7 @@
 , wheel
 
 # dependencies
+, beartype
 , einops
 , torch
 }:
@@ -29,6 +30,7 @@ buildPythonPackage rec {
   ];
 
   propagatedBuildInputs = [
+    beartype
     einops
     torch
   ];
@@ -42,6 +44,7 @@ buildPythonPackage rec {
   meta = with lib; {
     description = "Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch";
     homepage = "https://github.com/lucidrains/rotary-embedding-torch";
+    changelog = "https://github.com/lucidrains/rotary-embedding-torch/releases/tag/${version}";
     license = licenses.mit;
     maintainers = teams.tts.members;
   };