Unverified Commit 24896607 authored by Zexin Yuan's avatar Zexin Yuan
Browse files

python3Packages.warp-lang: fix unit tests

parent 2c7b73fe

pkgs/development/python-modules/warp-lang/default.nix

+12 −10
Original line number Diff line number Diff line
@@ -223,20 +223,22 @@ 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; })

      ];

      in

      writeShellApplication {

        name = "warp-lang-unit-tests";

        runtimeInputs = [ python' ];

        text = ''

        python3 -m warp.tests

          ${python'}/bin/python3 -m warp.tests

        '';

      };