nvidia-nemo--speech/nemo/collections/asr/inference/utils/endpointing_utils.py

# Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


def millisecond_to_frames(millisecond: int, ms_per_timestep: int) -> int:
    """
    Convert milliseconds to frames
    Args:
        millisecond (int): milliseconds to convert
        ms_per_timestep (int): milliseconds per timestep
    Returns:
        int: number of frames
    """
    return (millisecond + ms_per_timestep - 1) // ms_per_timestep


def get_custom_stop_history_eou(
    stop_history_eou: int | None, default_stop_history_eou: int, ms_per_timestep: int
) -> int:
    """
    Get the custom stop history of EOU
    Args:
        stop_history_eou (int): stop history of EOU
        default_stop_history_eou (int): default stop history of EOU
        ms_per_timestep (int): milliseconds per timestep
    Returns:
        int: custom stop history of EOU
    """
    if stop_history_eou is None:
        return default_stop_history_eou
    if stop_history_eou > 0:
        return millisecond_to_frames(stop_history_eou, ms_per_timestep)
    return 0 if stop_history_eou == 0 else -1