Skip to content

Commit c3cd0d7

Browse files
authored
Should be one underscore for module privates not two
1 parent c07f1d0 commit c3cd0d7

File tree

1 file changed

+18
-19
lines changed

1 file changed

+18
-19
lines changed

modules/script_callbacks.py

Lines changed: 18 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
4646

4747

4848
ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"])
49-
__callback_map = dict(
49+
_callback_map = dict(
5050
callbacks_app_started=[],
5151
callbacks_model_loaded=[],
5252
callbacks_ui_tabs=[],
@@ -58,19 +58,19 @@ def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
5858

5959

6060
def clear_callbacks():
61-
for callback_list in __callback_map.values():
61+
for callback_list in _callback_map.values():
6262
callback_list.clear()
6363

6464
def app_started_callback(demo: Optional[Blocks], app: FastAPI):
65-
for c in __callback_map['callbacks_app_started']:
65+
for c in _callback_map['callbacks_app_started']:
6666
try:
6767
c.callback(demo, app)
6868
except Exception:
6969
report_exception(c, 'app_started_callback')
7070

7171

7272
def model_loaded_callback(sd_model):
73-
for c in __callback_map['callbacks_model_loaded']:
73+
for c in _callback_map['callbacks_model_loaded']:
7474
try:
7575
c.callback(sd_model)
7676
except Exception:
@@ -80,7 +80,7 @@ def model_loaded_callback(sd_model):
8080
def ui_tabs_callback():
8181
res = []
8282

83-
for c in __callback_map['callbacks_ui_tabs']:
83+
for c in _callback_map['callbacks_ui_tabs']:
8484
try:
8585
res += c.callback() or []
8686
except Exception:
@@ -90,31 +90,31 @@ def ui_tabs_callback():
9090

9191

9292
def ui_settings_callback():
93-
for c in __callback_map['callbacks_ui_settings']:
93+
for c in _callback_map['callbacks_ui_settings']:
9494
try:
9595
c.callback()
9696
except Exception:
9797
report_exception(c, 'ui_settings_callback')
9898

9999

100100
def before_image_saved_callback(params: ImageSaveParams):
101-
for c in __callback_map['callbacks_before_image_saved']:
101+
for c in _callback_map['callbacks_before_image_saved']:
102102
try:
103103
c.callback(params)
104104
except Exception:
105105
report_exception(c, 'before_image_saved_callback')
106106

107107

108108
def image_saved_callback(params: ImageSaveParams):
109-
for c in __callback_map['callbacks_image_saved']:
109+
for c in _callback_map['callbacks_image_saved']:
110110
try:
111111
c.callback(params)
112112
except Exception:
113113
report_exception(c, 'image_saved_callback')
114114

115115

116116
def cfg_denoiser_callback(params: CFGDenoiserParams):
117-
for c in __callback_map['callbacks_cfg_denoiser']:
117+
for c in _callback_map['callbacks_cfg_denoiser']:
118118
try:
119119
c.callback(params)
120120
except Exception:
@@ -133,27 +133,27 @@ def remove_current_script_callbacks():
133133
filename = stack[0].filename if len(stack) > 0 else 'unknown file'
134134
if filename == 'unknown file':
135135
return
136-
for callback_list in __callback_map.values():
136+
for callback_list in _callback_map.values():
137137
for callback_to_remove in [cb for cb in callback_list if cb.script == filename]:
138138
callback_list.remove(callback_to_remove)
139139

140140

141141
def remove_callbacks_for_function(callback_func):
142-
for callback_list in __callback_map.values():
142+
for callback_list in _callback_map.values():
143143
for callback_to_remove in [cb for cb in callback_list if cb.callback == callback_func]:
144144
callback_list.remove(callback_to_remove)
145145

146146

147147
def on_app_started(callback):
148148
"""register a function to be called when the webui started, the gradio `Block` component and
149149
fastapi `FastAPI` object are passed as the arguments"""
150-
add_callback(__callback_map['callbacks_app_started'], callback)
150+
add_callback(_callback_map['callbacks_app_started'], callback)
151151

152152

153153
def on_model_loaded(callback):
154154
"""register a function to be called when the stable diffusion model is created; the model is
155155
passed as an argument"""
156-
add_callback(__callback_map['callbacks_model_loaded'], callback)
156+
add_callback(_callback_map['callbacks_model_loaded'], callback)
157157

158158

159159
def on_ui_tabs(callback):
@@ -166,35 +166,34 @@ def on_ui_tabs(callback):
166166
title is tab text displayed to user in the UI
167167
elem_id is HTML id for the tab
168168
"""
169-
add_callback(__callback_map['callbacks_ui_tabs'], callback)
169+
add_callback(_callback_map['callbacks_ui_tabs'], callback)
170170

171171

172172
def on_ui_settings(callback):
173173
"""register a function to be called before UI settings are populated; add your settings
174174
by using shared.opts.add_option(shared.OptionInfo(...)) """
175-
add_callback(__callback_map['callbacks_ui_settings'], callback)
175+
add_callback(_callback_map['callbacks_ui_settings'], callback)
176176

177177

178178
def on_before_image_saved(callback):
179179
"""register a function to be called before an image is saved to a file.
180180
The callback is called with one argument:
181181
- params: ImageSaveParams - parameters the image is to be saved with. You can change fields in this object.
182182
"""
183-
add_callback(__callback_map['callbacks_before_image_saved'], callback)
183+
add_callback(_callback_map['callbacks_before_image_saved'], callback)
184184

185185

186186
def on_image_saved(callback):
187187
"""register a function to be called after an image is saved to a file.
188188
The callback is called with one argument:
189189
- params: ImageSaveParams - parameters the image was saved with. Changing fields in this object does nothing.
190190
"""
191-
add_callback(__callback_map['callbacks_image_saved'], callback)
191+
add_callback(_callback_map['callbacks_image_saved'], callback)
192192

193193

194194
def on_cfg_denoiser(callback):
195195
"""register a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs.
196196
The callback is called with one argument:
197197
- params: CFGDenoiserParams - parameters to be passed to the inner model and sampling state details.
198198
"""
199-
add_callback(__callback_map['callbacks_cfg_denoiser'], callback)
200-
199+
add_callback(_callback_map['callbacks_cfg_denoiser'], callback)

0 commit comments

Comments
 (0)