Merge pull request #984 from Labelbox/kkim/fix-create-batch

Fix create batch

Author: kkim-labelbox

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+# Version 3.40.1 (2023-03-10)
+
+## Fixed
+* Fixed issue where calling create_batch() on exported data rows wasn't working
+
 # Version 3.40.0 (2023-03-10)
 
 ## Added

docs/source/conf.py

@@ -21,7 +21,7 @@
 copyright = '2021, Labelbox'
 author = 'Labelbox'
 
-release = '3.40.0'
+release = '3.40.1'
 
 # -- General configuration ---------------------------------------------------
 

labelbox/__init__.py

@@ -1,5 +1,5 @@
 name = "labelbox"
-__version__ = "3.40.0"
+__version__ = "3.40.1"
 
 from labelbox.client import Client
 from labelbox.schema.project import Project

labelbox/schema/project.py

@@ -814,7 +814,7 @@ def create_batch(
                         "`data_rows` must be DataRow ids or DataRow objects")
 
         if data_rows is not None:
-            row_count = len(data_rows)
+            row_count = len(dr_ids)
         elif global_keys is not None:
             row_count = len(global_keys)
         else: