diff --git a/pkgs/development/python-modules/einops/default.nix b/pkgs/development/python-modules/einops/default.nix
index 84ac540577c9..d240ce0c43d9 100644
--- a/pkgs/development/python-modules/einops/default.nix
+++ b/pkgs/development/python-modules/einops/default.nix
@@ -2,33 +2,37 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
   hatchling,
+
+  # tests
   jupyter,
   nbconvert,
   numpy,
   parameterized,
   pillow,
   pytestCheckHook,
-  writableTmpDirAsHomeHook,
   torch,
+  writableTmpDirAsHomeHook,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "einops";
-  version = "0.8.1";
+  version = "0.8.2";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "arogozhnikov";
     repo = "einops";
-    tag = "v${version}";
-    hash = "sha256-J9m5LMOleHf2UziUbOtwf+DFpu/wBDcAyHUor4kqrR8=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-d5Vbtkw/MChS2j2IC6j97wfVoKWZT9mU4OeXyEjm6ys=";
   };
 
   build-system = [ hatchling ];
 
   nativeCheckInputs = [
-    writableTmpDirAsHomeHook
     jupyter
     nbconvert
     numpy
@@ -36,6 +40,7 @@ buildPythonPackage rec {
     pillow
     pytestCheckHook
     torch
+    writableTmpDirAsHomeHook
   ];
 
   env.EINOPS_TEST_BACKENDS = "numpy";
@@ -51,10 +56,10 @@ buildPythonPackage rec {
   __darwinAllowLocalNetworking = true;
 
   meta = {
-    changelog = "https://github.com/arogozhnikov/einops/releases/tag/${src.tag}";
     description = "Flexible and powerful tensor operations for readable and reliable code";
     homepage = "https://github.com/arogozhnikov/einops";
+    changelog = "https://github.com/arogozhnikov/einops/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.mit;
     maintainers = with lib.maintainers; [ yl3dy ];
   };
-}
+})