From 2d4e5795030f2ceb8c94ca6f8ab326f05428f7b0 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Wed, 22 Jan 2025 10:41:23 -0800 Subject: [PATCH] Fix pylint error --- comfy/model_management_types.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/model_management_types.py b/comfy/model_management_types.py index 41add063e..25c8f917c 100644 --- a/comfy/model_management_types.py +++ b/comfy/model_management_types.py @@ -7,11 +7,12 @@ import torch import torch.nn from typing_extensions import TypedDict, NotRequired -from comfy.latent_formats import LatentFormat +from .latent_formats import LatentFormat ModelManageableT = TypeVar('ModelManageableT', bound='ModelManageable') LatentFormatT = TypeVar('LatentFormatT', bound=LatentFormat) + @runtime_checkable class DeviceSettable(Protocol): @property