Use INCLUDE_SELFTEST variable

2024-04-30 13:13:22 -06:00
parent 725d3aeb58
commit d12cb9e96b
1 changed files with 1 additions and 1 deletions
@@ -410,7 +410,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
    "KRestartSamplerCustom": "KSampler With Restarts (Custom)",
 }

-if os.environ.get("COMFYUI_RESTART_SAMPLING_SELFTEST", "").strip() == "1":
+if INCLUDE_SELFTEST:

    class RestartSelfTestNode:
        @classmethod