1 files changed, 5 insertions, 0 deletions
diff --git a/infra/libkookie/nixpkgs/pkgs/development/python-modules/pymc3/default.nix b/infra/libkookie/nixpkgs/pkgs/development/python-modules/pymc3/default.nix
index bda74c5063a6..dfbfe90d2421 100644
--- a/infra/libkookie/nixpkgs/pkgs/development/python-modules/pymc3/default.nix
+++ b/infra/libkookie/nixpkgs/pkgs/development/python-modules/pymc3/default.nix
@@ -14,6 +14,8 @@
 , pytest
 , nose
 , parameterized
+, fastprogress
+, typing-extensions
 }:
 
 buildPythonPackage rec {
@@ -41,6 +43,8 @@ buildPythonPackage rec {
     h5py
     arviz
     packaging
+    fastprogress
+    typing-extensions
   ];
 
   checkInputs = [
@@ -52,6 +56,7 @@ buildPythonPackage rec {
   # The test suite is computationally intensive and test failures are not
   # indicative for package usability hence tests are disabled by default.
   doCheck = false;
+  pythonImportsCheck = [ "pymc3" ];
 
   # For some reason tests are run as a part of the *install* phase if enabled.
   # Theano writes compiled code to ~/.theano hence we set $HOME.