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

kkim-labelbox · web-flow · commit 8ffa9da882d8 · 2023-03-10T14:30:10.000-08:00
Fix create batch
diff --git a/CHANGELOG.md b/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
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -21,7 +21,7 @@
 copyright = '2021, Labelbox'
 author = 'Labelbox'
 
-release = '3.40.0'
+release = '3.40.1'
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/labelbox/__init__.py b/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
diff --git a/labelbox/schema/project.py b/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:
