From 9549e3fe7aa9226434034498676d55934f8a67fb Mon Sep 17 00:00:00 2001 From: pydn Date: Fri, 25 Aug 2023 11:29:09 -0500 Subject: [PATCH] Updated docstring.: --- comfyui_to_python.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/comfyui_to_python.py b/comfyui_to_python.py index 954b0db..33e6e53 100644 --- a/comfyui_to_python.py +++ b/comfyui_to_python.py @@ -348,6 +348,12 @@ class CodeGenerator: def clean_variable_name(class_type: str) -> str: """ Remove any characters from variable name that could cause errors running the Python script. + + Args: + class_type (str): Class type. + + Returns: + str: Cleaned variable name with no special characters or spaces """ # Convert to lowercase and replace spaces with underscores clean_name = class_type.lower().strip().replace("-", "_").replace(" ", "_")