Error with Canny pre-processor

[Fusedon]

Traceback (most recent call last):
File "/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/cog/server/worker.py", line 222, in _predict
for r in result:
File "/root/.pyenv/versions/3.9.18/lib/python3.9/site-packages/torch/utils/_contextlib.py", line 35, in generator_context
response = gen.send(None)
File "predict.py", line 338, in predict
pipe, kwargs = self.build_pipe(
File "predict.py", line 186, in build_pipe
img = self.canny_preprocess(img, low_threshold, high_threshold)
TypeError: canny_preprocess() takes 2 positional arguments but 4 were given