numpy deserialization

Author: miguelgrinberg

elasticsearch/dsl/field.py

@@ -1555,6 +1555,7 @@ class DenseVector(Field):
     :arg dynamic:
     :arg fields:
     :arg synthetic_source_keep:
+    :arg use_numpy: if set to ``True``, deserialize as a numpy array.
     """
 
     name = "dense_vector"
@@ -1587,6 +1588,7 @@ def __init__(
         synthetic_source_keep: Union[
             Literal["none", "arrays", "all"], "DefaultType"
         ] = DEFAULT,
+        use_numpy: bool = False,
         **kwargs: Any,
     ):
         if dims is not DEFAULT:
@@ -1614,6 +1616,7 @@ def __init__(
         self._element_type = kwargs.get("element_type", "float")
         if self._element_type in ["float", "byte"]:
             kwargs["multi"] = True
+        self._use_numpy = use_numpy
         super().__init__(*args, **kwargs)
 
 

utils/templates/field.py.tpl

@@ -217,6 +217,9 @@ class {{ k.name }}({{ k.parent }}):
             {% endfor %}
         {% endfor %}
     {% endif %}
+    {% if k.field == "dense_vector" %}
+    :arg use_numpy: if set to ``True``, deserialize as a numpy array.
+    {% endif %}
     """
     name = "{{ k.field }}"
     {% if k.coerced %}
@@ -246,6 +249,9 @@ class {{ k.name }}({{ k.parent }}):
         {{ arg.name }}: {{ arg.type }} = DEFAULT,
             {% endif %}
         {% endfor %}
+        {% if k.field == "dense_vector" %}
+        use_numpy: bool = False,
+        {% endif %}
         **kwargs: Any
     ):
         {% for arg in k.args %}
@@ -416,6 +422,7 @@ class {{ k.name }}({{ k.parent }}):
         self._element_type = kwargs.get("element_type", "float")
         if self._element_type in ["float", "byte"]:
             kwargs["multi"] = True
+        self._use_numpy = use_numpy
         super().__init__(*args, **kwargs)
 
 class NumpyDenseVector(DenseVector):