Files

10 lines
347 B
Python

import sys
from unittest.mock import MagicMock
# Mock comfy_api before any project module is imported.
# io.ComfyNode must be a real class so ConstrainResolution can inherit from it.
_comfy_api_latest = MagicMock()
_comfy_api_latest.io.ComfyNode = object
sys.modules['comfy_api'] = MagicMock()
sys.modules['comfy_api.latest'] = _comfy_api_latest