def test_array_set_prefix_path():
arr = Array(
storage_type="lilcom_hdf5",
storage_path="data/train",
storage_key="irrelevant",
shape=[300],
)
arr1 = arr.with_path_prefix("/newhome")
assert arr1.storage_path == "/newhome/data/train"
assert arr1.storage_type == arr.storage_type
assert arr1.storage_key == arr.storage_key
assert arr1.shape == arr.shape
def test_array_serialization():
# Individual items do not support JSON/etc. serialization;
# instead, the XSet (e.g. CutSet) classes convert them to dicts.
manifest = Array(
storage_type="lilcom_hdf5",
storage_path="/tmp/data",
storage_key="irrelevant",
shape=[300],
)
serialized = manifest.to_dict()
restored = Array.from_dict(serialized)
assert manifest == restored
def test_cut_move_to_memory_load_custom_false():
path = "test/fixtures/libri/cuts.json"
cut = CutSet.from_file(path)[0]
cut.custom_array = Array("irrelevant", "irrelevant", "irrelevant", [10])
cut_mem = cut.move_to_memory(load_custom=False)
assert cut.custom_array == cut_mem.custom_array # nothing was copied
def store_array(
self,
key: str,
value: np.ndarray,
frame_shift: Optional[Seconds] = None,
temporal_dim: Optional[int] = None,
start: Seconds = 0,
) -> Union[Array, TemporalArray]:
"""
Store a numpy array in the underlying storage and return a manifest
describing how to retrieve the data.
If the array contains a temporal dimension (e.g. it represents the
frame-level features, alignment, posteriors, etc. of an utterance)
then ``temporal_dim`` and ``frame_shift`` may be specified to enable
downstream padding, truncating, and partial reads of the array.
:param key: An ID that uniquely identifies the array.
:param value: The array to be stored.
:param frame_shift: Optional float, when the array has a temporal dimension
it indicates how much time has passed between the starts of consecutive frames
(expressed in seconds).
:param temporal_dim: Optional int, when the array has a temporal dimension,
it indicates which dim to interpret as temporal.
:param start: Float, when the array is temporal, it indicates what is the offset
of the array w.r.t. the start of recording. Useful for reading subsets
of an array when it represents something computed from long recordings.
Ignored for non-temporal arrays.
:return: A manifest of type :class:`~lhotse.array.Array` or
:class:`~lhotse.array.TemporalArray`, depending on the input arguments.
"""
is_temporal = frame_shift is not None and temporal_dim is not None
if not is_temporal:
assert all(arg is None for arg in [frame_shift, temporal_dim]), (
"frame_shift and temporal_dim have to be both None or both set "
f"(got frame_shift={frame_shift}, temporal_dim={temporal_dim})."
)
storage_key = self.write(key, value)
array = Array(
storage_type=self.name,
storage_path=self.storage_path,
storage_key=storage_key,
shape=list(value.shape),
)
if not is_temporal:
return array
return TemporalArray(
array=array,
temporal_dim=temporal_dim,
frame_shift=frame_shift,
start=start,
)
def test_temporal_array_set_prefix_path():
arr = TemporalArray(
array=Array(
storage_type="lilcom_hdf5",
storage_path="data/train",
storage_key="irrelevant",
shape=[300],
),
temporal_dim=0,
frame_shift=0.3,
start=5.0,
)
arr1 = arr.with_path_prefix("/newhome")
assert arr1.array.storage_path == "/newhome/data/train"
assert arr1.array.storage_type == arr.array.storage_type
assert arr1.array.storage_key == arr.array.storage_key
assert arr1.shape == arr.shape
assert arr1.temporal_dim == arr.temporal_dim
assert arr1.frame_shift == arr.frame_shift
assert arr1.start == arr.start
def validate_array(arr: Array, read_data: bool = False) -> None:
if read_data:
data = arr.load()
assert data.shape == arr.shape