python312Packages.blosc2: make torch tests optional, enabled by default (#383173)

2025-02-18 23:59:02 +01:00
parent 47bcb8da9b 5cf57e1969
commit 7cfecb1fec
1 changed files with 4 additions and 1 deletions
@@ -1,5 +1,6 @@
 {
  lib,
+  stdenv,
  buildPythonPackage,
  fetchFromGitHub,

@@ -24,6 +25,8 @@
  # tests
  psutil,
  pytestCheckHook,
+  torch,
+  runTorchTests ? lib.meta.availableOn stdenv.hostPlatform torch,
 }:

 buildPythonPackage rec {
@@ -68,7 +71,7 @@ buildPythonPackage rec {
  nativeCheckInputs = [
    psutil
    pytestCheckHook
-  ];
+  ] ++ lib.optionals runTorchTests [ torch ];

  disabledTests = [
    # RuntimeError: Error while getting the slice