Coverage for tests/fantasytalking_mock.py: 100%

1""" 

2A mock for Fantasy Talking modules used in testing. 

3""" 

4 

5from unittest.mock import MagicMock 

6 

7from typing import Dict 

8from typing import Any 

9 

10 

11class FantasyTalkingMock(MagicMock): 

12 

13 def get_sub_modules(self) -> Dict[str, Any]: 

14 return { 

15 "torchvision": MagicMock(), 

16 "torchvision.transforms": MagicMock(), 

17 "torchvision.transforms.functional": MagicMock(), 

18 "xfuser": MagicMock(), 

19 "xfuser.config": MagicMock(), 

20 "xfuser.core": MagicMock(), 

21 "xfuser.core.distributed": MagicMock(), 

22 "diffsynth": MagicMock(), 

23 "diffsynth.models": MagicMock(), 

24 "diffsynth.models.wan_video_dit": MagicMock(), 

25 "wan.distributed": MagicMock(), 

26 "wan.distributed.xdit_context_parallel": MagicMock(), 

27 "wan.distributed.fsdp": MagicMock(), 

28 "transformers": MagicMock(), 

29 "model": MagicMock(), 

30 "utils": MagicMock(), 

31 }