1 files changed, 9 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/labelbox/default.nix b/pkgs/development/python-modules/labelbox/default.nix
index b761b71685331..063a223cbd5a6 100644
--- a/pkgs/development/python-modules/labelbox/default.nix
+++ b/pkgs/development/python-modules/labelbox/default.nix
@@ -6,6 +6,9 @@
 , pillow
 , rasterio
 , shapely
+, ndjson
+, backoff
+, google_api_core
 }:
 
 buildPythonPackage rec {
@@ -17,11 +20,16 @@ buildPythonPackage rec {
     sha256 = "2be6c03dafce0a786cfab5d120196efccaf300cab5aee4d2fdad644b7bee1aef";
   };
 
-  propagatedBuildInputs = [ jinja2 requests pillow rasterio shapely ];
+  propagatedBuildInputs = [
+    jinja2 requests pillow rasterio shapely ndjson backoff
+    google_api_core
+  ];
 
   # Test cases are not running on pypi or GitHub
   doCheck = false;
 
+  pythonImportsCheck = [ "labelbox" ];
+
   meta = with lib; {
     homepage = "https://github.com/Labelbox/Labelbox";
     description = "Platform API for LabelBox";