diff --git a/pkgs/development/python-modules/pytorch-metric-learning/default.nix b/pkgs/development/python-modules/pytorch-metric-learning/default.nix
index dd6ce14adf10..af57233b9cf2 100644
--- a/pkgs/development/python-modules/pytorch-metric-learning/default.nix
+++ b/pkgs/development/python-modules/pytorch-metric-learning/default.nix
@@ -27,7 +27,7 @@
 
 buildPythonPackage rec {
   pname = "pytorch-metric-learning";
-  version = "2.7.0";
+  version = "2.8.1";
   pyproject = true;
 
   disabled = isPy27;
@@ -35,8 +35,8 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "KevinMusgrave";
     repo = pname;
-    rev = "refs/tags/v${version}";
-    hash = "sha256-mxAl4GMyAWtvocc68Ac3z1+W13k9OOK7aQFfB7X0f9c=";
+    tag = "v${version}";
+    hash = "sha256-WO/gv8rKkxY3pR627WrEPVyvZnvUZIKMzOierIW8bJA=";
   };
 
   build-system = [