summary refs log tree commit diff
path: root/pkgs/development/python-modules/torch-tb-profiler/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/torch-tb-profiler/default.nix')
-rw-r--r--pkgs/development/python-modules/torch-tb-profiler/default.nix2
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/torch-tb-profiler/default.nix b/pkgs/development/python-modules/torch-tb-profiler/default.nix
index 41ff63a21fa..108acfb0d35 100644
--- a/pkgs/development/python-modules/torch-tb-profiler/default.nix
+++ b/pkgs/development/python-modules/torch-tb-profiler/default.nix
@@ -27,7 +27,7 @@ buildPythonPackage rec {
 
   propagatedBuildInputs = [ pandas tensorboard ];
 
-  checkInputs = [ pytestCheckHook torch torchvision ];
+  nativeCheckInputs = [ pytestCheckHook torch torchvision ];
 
   disabledTests = [
     # Tests that attempt to access the filesystem in naughty ways.