docarray/docarray/typing/tensor/audio/audio_ndarray.py at main · docarray/docarray

This repository was archived by the owner on Sep 9, 2026. It is now read-only.

Repository navigation

from docarray.typing.proto_register import _register_proto

from docarray.typing.tensor.audio.abstract_audio_tensor import AbstractAudioTensor

from docarray.typing.tensor.ndarray import NdArray

@_register_proto(proto_type_name='audio_ndarray')

class AudioNdArray(AbstractAudioTensor, NdArray):

"""

Subclass of [`NdArray`][docarray.typing.NdArray], to represent an audio tensor.

Adds audio-specific features to the tensor.

---

```python

from typing import Optional

from docarray import BaseDoc

from docarray.typing import AudioBytes, AudioNdArray, AudioUrl

import numpy as np

class MyAudioDoc(BaseDoc):

title: str

audio_tensor: Optional[AudioNdArray] = None

url: Optional[AudioUrl] = None

bytes_: Optional[AudioBytes] = None

# from tensor

doc_1 = MyAudioDoc(

title='my_first_audio_doc',

audio_tensor=np.random.rand(1000, 2),

)

# doc_1.audio_tensor.save(file_path='/tmp/file_1.wav')

doc_1.bytes_ = doc_1.audio_tensor.to_bytes()

# from url

doc_2 = MyAudioDoc(

title='my_second_audio_doc',

url='https://www.kozco.com/tech/piano2.wav',

)

doc_2.audio_tensor, _ = doc_2.url.load()

# doc_2.audio_tensor.save(file_path='/tmp/file_2.wav')

doc_2.bytes_ = doc_1.audio_tensor.to_bytes()

```

---

"""

...