llama-stack-mirror/llama_stack/apis/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
..
__init__.py API Updates (#73) 2024-09-17 19:51:35 -07:00
post_training.py chore: fix mypy violations in post_training modules 2025-03-17 19:12:09 -04:00