llama-stack-mirror/llama_stack/providers/inline/post_training/torchtune
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
datasets chore: fix mypy violations in post_training modules 2025-03-17 19:12:09 -04:00
recipes chore: fix mypy violations in post_training modules 2025-03-17 19:12:09 -04:00
__init__.py chore: fix typing hints for get_provider_impl deps arguments (#1544) 2025-03-11 10:07:28 -07:00
config.py test: add unit test to ensure all config types are instantiable (#1601) 2025-03-12 22:29:58 -07:00
post_training.py refactor: move all datetime.now() calls to UTC (#1589) 2025-03-13 15:34:53 -07:00