diff --git a/pkgs/development/python-modules/skorch/default.nix b/pkgs/development/python-modules/skorch/default.nix
index e3380eb8166e..82e9d14614f7 100644
--- a/pkgs/development/python-modules/skorch/default.nix
+++ b/pkgs/development/python-modules/skorch/default.nix
@@ -1,35 +1,40 @@
 {
   lib,
-  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
+  setuptools,
+
+  # dependencies
   numpy,
+  pandas,
   scikit-learn,
   scipy,
-  setuptools,
   tabulate,
   torch,
   tqdm,
+
+  # tests
   flaky,
-  llvmPackages,
   openssl,
-  pandas,
   pytest-cov-stub,
   pytestCheckHook,
   safetensors,
   transformers,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "skorch";
   version = "1.4.0";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "skorch-dev";
     repo = "skorch";
-    tag = "v${version}";
-    sha256 = "sha256-il3S5cfW47tKvMQGr/BfbEjMEMVzBF4gSrQhR1uKxks=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-il3S5cfW47tKvMQGr/BfbEjMEMVzBF4gSrQhR1uKxks=";
   };
 
   build-system = [ setuptools ];
@@ -75,6 +80,7 @@ buildPythonPackage rec {
     "skorch/tests/test_dataset.py"
     "skorch/tests/test_hf.py"
     "skorch/tests/llm/test_llm_classifier.py"
+
     # These tests fail when running in parallel for all platforms with:
     # "RuntimeError: The server socket has failed to listen on any local
     # network address because they use the same hardcoded port."
@@ -91,4 +97,4 @@ buildPythonPackage rec {
     license = lib.licenses.bsd3;
     maintainers = with lib.maintainers; [ bcdarwin ];
   };
-}
+})