From afbb8aa1548436fca8534ffc31a8fa329fd0024d Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Fri, 4 Oct 2024 21:10:19 -0700 Subject: [PATCH] Fix #23 --- comfy/component_model/folder_path_types.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/comfy/component_model/folder_path_types.py b/comfy/component_model/folder_path_types.py index dd75080b5..ac8750935 100644 --- a/comfy/component_model/folder_path_types.py +++ b/comfy/component_model/folder_path_types.py @@ -75,6 +75,9 @@ class FolderNames: def keys(self): return self.contents.keys() + def get(self, key, __default=None): + return self.contents.get(key, __default) + class SaveImagePathResponse(NamedTuple): full_output_folder: str