1 files changed, 5 insertions, 1 deletions
diff --git a/infra/libkookie/nixpkgs/pkgs/development/python-modules/Theano/default.nix b/infra/libkookie/nixpkgs/pkgs/development/python-modules/Theano/default.nix
index cc2b681e415d..ce32e42e4716 100644
--- a/infra/libkookie/nixpkgs/pkgs/development/python-modules/Theano/default.nix
+++ b/infra/libkookie/nixpkgs/pkgs/development/python-modules/Theano/default.nix
@@ -68,7 +68,9 @@ in buildPythonPackage rec {
       --replace 'StrParam(default_dnn_base_path)' 'StrParam('\'''${cudnn}'\''')'
   '';
 
-  preCheck = ''
+  # needs to be postFixup so it runs before pythonImportsCheck even when
+  # doCheck = false (meaning preCheck would be disabled)
+  postFixup = ''
     mkdir -p check-phase
     export HOME=$(pwd)/check-phase
   '';
@@ -81,6 +83,8 @@ in buildPythonPackage rec {
   checkInputs = [ nose ];
   propagatedBuildInputs = [ numpy numpy.blas scipy six libgpuarray_ ];
 
+  pythonImportsCheck = [ "theano" ];
+
   meta = with stdenv.lib; {
     homepage = "http://deeplearning.net/software/theano/";
     description = "A Python library for large-scale array computation";