feat: batch_size value must be less than 40

TremaMiguel · TremaMiguel · commit e52a895e0438 · 2022-03-07T18:07:12.000-06:00
Signed-off-by: Miguel Trejo &lt;armando.trejo.marrufo@gmail.com&gt;
diff --git a/sdk/python/feast/infra/online_stores/dynamodb.py b/sdk/python/feast/infra/online_stores/dynamodb.py
@@ -188,7 +188,7 @@ def online_read(
         table: FeatureView,
         entity_keys: List[EntityKeyProto],
         requested_features: Optional[List[str]] = None,
-        batch_size: int = 10,
+        batch_size: int = 20,
     ) -> List[Tuple[Optional[datetime], Optional[Dict[str, ValueProto]]]]:
         """
         Retrieve feature values from the online DynamoDB store.
@@ -205,6 +205,10 @@ def online_read(
                 to set batch_size value less than 40 to avoid ``UnprocessedKeys`` and
                 ``ValidationException`` errors.
         """
+        if batch_size > 40:
+            raise ValueError(
+                f"batch_size value must be less than 40, input value is {batch_size}"
+            )
         online_config = config.online_store
         assert isinstance(online_config, DynamoDBOnlineStoreConfig)
         dynamodb_resource = self._get_dynamodb_resource(online_config.region)
