Doesn't work with Upscaler Node

[ZenRepublic]

this the error:

Traceback (most recent call last):
  File "C:\pinokio\api\comfy.git\app\custom_nodes\comfyui-to-python-extension\__init__.py", line 50, in save_as_script
    ComfyUItoPython(workflow=workflow, output_file=sio)
  File "C:\pinokio\api\comfy.git\app\custom_nodes\comfyui-to-python-extension\comfyui_to_python.py", line 544, in __init__
    self.execute()
  File "C:\pinokio\api\comfy.git\app\custom_nodes\comfyui-to-python-extension\comfyui_to_python.py", line 573, in execute
    generated_code = code_generator.generate_workflow(
  File "C:\pinokio\api\comfy.git\app\custom_nodes\comfyui-to-python-extension\comfyui_to_python.py", line 290, in generate_workflow
    final_code = self.assemble_python_code(
  File "C:\pinokio\api\comfy.git\app\custom_nodes\comfyui-to-python-extension\comfyui_to_python.py", line 413, in assemble_python_code
    final_code = black.format_str(final_code, mode=black.Mode())
  File "src\black\__init__.py", line 1204, in format_str
  File "src\black\__init__.py", line 1218, in _format_str_once
  File "src\black\parsing.py", line 98, in lib2to3_parse
black.parsing.InvalidInput: Cannot parse: 181:116:                      imageupscalewithmodel_42:1 = imageupscalewithmodel.upscale(upscale_model=get_value_at_index(upscalemodelloader_42:0, 0), image=get_value_at_index(vaedecode_33, 0))

removing the upscaler nodes from workflow exports correctly

btw works with default comfy core upscaler

[mbylstra]

You can try: #114