diff --git a/python/sparknlp/common.py b/python/sparknlp/common.py
index <HASH>..<HASH> 100644
--- a/python/sparknlp/common.py
+++ b/python/sparknlp/common.py
@@ -88,6 +88,17 @@ class HasStorageRef:
         return self.getOrDefault("storageRef")
 
 
+class HasBatchedAnnotate:
+
+    batchSize = Param(Params._dummy(), "batchSize", "Size of every batch", TypeConverters.toInt)
+
+    def setBatchSize(self, v):
+        return self._set(batchSize=v)
+
+    def getBatchSize(self):
+        return self.getOrDefault("batchSize")
+
+
 class HasCaseSensitiveProperties:
     caseSensitive = Param(Params._dummy(),
                           "caseSensitive",
Add HasBatchedAnnotate class to Python [skip ci]