diff --git a/nixnet/_session/signals.py b/nixnet/_session/signals.py index ed31b228..cf621a7c 100644 --- a/nixnet/_session/signals.py +++ b/nixnet/_session/signals.py @@ -4,6 +4,8 @@ import typing # NOQA: F401 +import six + from nixnet import _funcs from nixnet import _props @@ -49,7 +51,7 @@ def read(self): """ num_signals = len(self) timestamps, values = _funcs.nx_read_signal_single_point(self._handle, num_signals) - for timestamp, value in zip(timestamps, values): + for timestamp, value in six.moves.zip(timestamps, values): yield timestamp.value, value.value diff --git a/nixnet/convert.py b/nixnet/convert.py index b1242066..6da45fb1 100644 --- a/nixnet/convert.py +++ b/nixnet/convert.py @@ -6,6 +6,8 @@ import typing # NOQA: F401 import warnings +import six + from nixnet import _frames from nixnet import _funcs from nixnet import _props @@ -176,7 +178,7 @@ def _convert_bytes_to_signals(self, bytes): # type: (bytes) -> typing.Iterable[typing.Tuple[int, float]] num_signals = len(self.signals) timestamps, values = _funcs.nx_convert_frames_to_signals_single_point(self._handle, bytes, num_signals) - for timestamp, value in zip(timestamps, values): + for timestamp, value in six.moves.zip(timestamps, values): yield timestamp.value, value.value def convert_frames_to_signals(self, frames):