o "i@s@ddlZddlmZddlmZmZmZmZGdddeZdS)N)Dict)TrainingPlatformTrainingConfig JobStatusTrainingResultc@s\eZdZddZdedefddZdedefdd Zdede fd d Z dede fd d Z dS) MockPlatformcCs i|_dS)Njobs)selfr U/lsinfo/ai/hellotax_ai/training_center/backend/app/services/training_platform/mock.py__init__s zMockPlatform.__init__configreturncCs$tt}|dddd|j|<|S)Ngrunningr)rprogressstatuschecks)struuiduuid4r )r rjob_idr r r create_training_job s  z MockPlatform.create_training_jobrcCs||jvr td|d|j|}|dd7<t|ddd|d<|ddkr7d|d <tdd d d d Std|dd|ddd|ddd S)NJob not foundrr dr completedrgY@zTraining completed successfullyzMock training logs)rrmessagelogsrzTraining in progress: %zMock training logs at )r ValueErrorminrr rjobr r r get_job_statuss&   zMockPlatform.get_job_statuscCsl||jvr td|d|j|}|ddkr td|dtd|dddd d d d |id S)Nrrrrz not completed yet mock_model_g?gffffff?)lossaccuracy model_pathz /mock/models/)model_idmetrics artifacts)r r"rr$r r r get_job_result,s    zMockPlatform.get_job_resultcCs ||jvrdSd|j|d<dS)NF cancelledrTr )r rr r r cancel_job:s zMockPlatform.cancel_jobN) __name__ __module__ __qualname__rrrrrr&rr/boolr1r r r r rs  r) rtypingrbaserrrrrr r r r s