]jzddlZddlmZddlmZddlmZddlmZGddejjZ y)N) load_func)settings)timezone)ModelRuncHeZdZdZej eejZ ejdZ ejdZ dZ ejj Zej$j&ZdZdZdZdZdZdZGddZej8dZy) ModelRunFactoryz Factory for `ml_models.models.ModelRun`. Defaults are intentionally minimal; override fields in tests as needed. zproject.organizationzproject.created_byNrceZdZeZy)ModelRunFactory.MetaN)__name__ __module__ __qualname__rmodelP/root/env/lib/python3.12/site-packages/label_studio/ml_models/tests/factories.pyMetar srrc |sy|jtjjk(rOt j }|j ||_|j||_|jddgyy)zR Ensure timestamps are set when status implies they should exist. N triggered_at completed_at) update_fields) statusrModelRunStatus COMPLETEDrnowrrsave)selfcreate extractedkwargsrs r_default_timestampsz#ModelRunFactory._default_timestamps"sm   ;;(11;; ;,,.C  ($'!  ($'! II^^$DI E r9s,' !%(Fgnn77(Fr