fix: Add missing shutdown handler for TorchtunePostTrainingImpl

Without it, when server shuts down, it posts the following warning:

__main__:129 server: No shutdown method for TorchtunePostTrainingImpl

Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
This commit is contained in:
Ihar Hrachyshka 2025-03-11 00:33:18 +00:00
parent bc8daf7fea
commit f7c1e87293

View file

@ -43,6 +43,9 @@ class TorchtunePostTrainingImpl:
self.jobs = {}
self.checkpoints_dict = {}
async def shutdown(self):
pass
async def supervised_fine_tune(
self,
job_uuid: str,