llama-stack-mirror/llama_stack/providers/inline/post_training
Ihar Hrachyshka 8c01246344 chore: fix mypy violations in post_training modules
Note: this patch touches all files but post_training.py that will be
significantly changed by #1437, hence leaving it out of the picture for
now.

running_loss is now always Tensor (on-device) and doesn't change its
type from int to Tensor (which made mypy unhappy).

Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
2025-03-17 19:12:09 -04:00
..
common chore: fix mypy violations in post_training modules 2025-03-17 19:12:09 -04:00
torchtune chore: fix mypy violations in post_training modules 2025-03-17 19:12:09 -04:00
__init__.py Add init files to post training folders (#711) 2025-01-13 20:19:18 -08:00