[BUG] Incorrect documentation for Continuous Flow matching augmentation #1272
Description
BioNeMo Framework Version
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