| @@ -20,7 +20,18 @@ class Mock(MagicMock): | |||||
| return MagicMock() | return MagicMock() | ||||
| MOCK_MODULES = ["numpy", "pyswip", "torch", "torchvision", "zoopt", "termcolor"] | |||||
| MOCK_MODULES = [ | |||||
| "numpy", | |||||
| "pyswip", | |||||
| "torch", | |||||
| "torchvision", | |||||
| "zoopt", | |||||
| "termcolor", | |||||
| "torch.nn", | |||||
| "torch.utils", | |||||
| "torch.optim", | |||||
| "torchvision.transform", | |||||
| ] | |||||
| sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES) | sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES) | ||||