python3Packages.warp-lang: fix unit tests

2025-11-14 19:53:31 +08:00
parent 2c7b73feaf
commit 248966073f
1 changed files with 12 additions and 10 deletions
@@ -223,22 +223,24 @@ buildPythonPackage {
    inherit libmathdx;

    # Scripts which provide test packages and implement test logic.
-    testers.unit-tests = writeShellApplication {
-      name = "warp-lang-unit-tests";
-      runtimeInputs = [
+    testers.unit-tests =
+      let
        # Use the references from args
-        (python.withPackages (_: [
+        python' = python.withPackages (_: [
          warp-lang
          jax
          torch
-        ]))
+        ]);
        # Disable paddlepaddle interop tests: malloc(): unaligned tcache chunk detected
        #  (paddlepaddle.override { inherit cudaSupport; })
-      ];
-      text = ''
-        python3 -m warp.tests
-      '';
-    };
+      in
+      writeShellApplication {
+        name = "warp-lang-unit-tests";
+        runtimeInputs = [ python' ];
+        text = ''
+          ${python'}/bin/python3 -m warp.tests
+        '';
+      };

    # Tests run within the Nix sandbox.
    tests =