def update_source_segment()

in simuleval/states/states.py [0:0]

• 14 lines of code
• 5 McCabe index (conditional complexity)

    def update_source_segment(self, num_segment=1):
        # Read a segment from server
        info = self.get_info_from_server(num_segment)
        segment = info["segment"]

        self.segments.source.append(segment)

        if (
            info.get("finish", False) is True
            or segment in [DEFAULT_EOS]
        ):
            self.status["read"] = False
            # Receive an EOS from server
            if segment in [DEFAULT_EOS]:
                return

        # Preprocess a segment into units
        units = self.segment_to_units(segment)

        # Update the source unit entry
        for unit in units:
            self.unit_queue.source.push(unit)