diff --git a/pkgs/development/python-modules/onnx/default.nix b/pkgs/development/python-modules/onnx/default.nix
index f0fa2543f0b8..dfdd52372c14 100644
--- a/pkgs/development/python-modules/onnx/default.nix
+++ b/pkgs/development/python-modules/onnx/default.nix
@@ -15,13 +15,17 @@
 
   # dependencies
   numpy,
+  typing-extensions,
 
+  # tests
   google-re2,
+  ml-dtypes,
   nbval,
   parameterized,
   pillow,
   pytestCheckHook,
   tabulate,
+  writableTmpDirAsHomeHook,
 }:
 
 let
@@ -29,14 +33,14 @@ let
 in
 buildPythonPackage rec {
   pname = "onnx";
-  version = "1.17.0";
+  version = "1.18.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "onnx";
     repo = "onnx";
     tag = "v${version}";
-    hash = "sha256-9oORW0YlQ6SphqfbjcYb0dTlHc+1gzy9quH/Lj6By8Q=";
+    hash = "sha256-UhtF+CWuyv5/Pq/5agLL4Y95YNP63W2BraprhRqJOag=";
   };
 
   build-system = [
@@ -54,15 +58,18 @@ buildPythonPackage rec {
   dependencies = [
     protobuf
     numpy
+    typing-extensions
   ];
 
   nativeCheckInputs = [
     google-re2
+    ml-dtypes
     nbval
     parameterized
     pillow
     pytestCheckHook
     tabulate
+    writableTmpDirAsHomeHook
   ];
 
   postPatch = ''
@@ -92,11 +99,9 @@ buildPythonPackage rec {
   # The setup.py does all the configuration
   dontUseCmakeConfigure = true;
 
+  # detecting source dir as a python package confuses pytest
   preCheck = ''
-    export HOME=$(mktemp -d)
-
-    # detecting source dir as a python package confuses pytest
-    mv onnx/__init__.py onnx/__init__.py.hidden
+    rm onnx/__init__.py
   '';
 
   pytestFlagsArray = [