[BUG] Incorrect documentation for Continuous Flow matching augmentation

[metric-space]

BioNeMo Framework Version

c18ce7a

Bug Description

Documentation here is wrong:
https://github.com/NVIDIA/bionemo-framework/blob/main/sub-packages/bionemo-moco/src/bionemo/moco/interpolants/continuous_time/continuous/continuous_flow_matching.py#L56

apply_augmentation doesn't output time for any of the three augmentation strategies

• https://github.com/NVIDIA/bionemo-framework/blob/main/sub-packages/bionemo-moco/src/bionemo/moco/interpolants/continuous_time/continuous/data_augmentation/ot_sampler.py#L209
• https://github.com/NVIDIA/bionemo-framework/blob/main/sub-packages/bionemo-moco/src/bionemo/moco/interpolants/continuous_time/continuous/data_augmentation/kabsch_augmentation.py#L148
• https://github.com/NVIDIA/bionemo-framework/blob/main/sub-packages/bionemo-moco/src/bionemo/moco/interpolants/continuous_time/continuous/data_augmentation/equivariant_ot_sampler.py#L243

Steps to Reproduce

It's documentation

Error Messages and Logs

Docker Image

No response

System Information

Environment Details:

• OS: [e.g., Ubuntu 20.04]
• CPU: [e.g., Intel i9-12900K]
• RAM: [e.g., 64GB]

GPU Details:

• GPU Model: [e.g., NVIDIA RTX 4090]
• GPU Memory: [e.g., 24GB]
• CUDA Version: [e.g., 12.1]
• CUDA Driver: [e.g., 525.85.05]
• cuDNN Version: [e.g., 8.9.0]

Additional Context

No response