from . import smea_sampling
from .smea_sampling import sample_euler_dy, sample_euler_smea_dy, sample_euler_negative, sample_euler_dy_negative

if smea_sampling.BACKEND == "ComfyUI":
    if not smea_sampling.INITIALIZED:
        from comfy.k_diffusion import sampling as k_diffusion_sampling
        from comfy.samplers import SAMPLER_NAMES

        setattr(k_diffusion_sampling, "sample_euler_dy", sample_euler_dy)
        setattr(k_diffusion_sampling, "sample_euler_smea_dy", sample_euler_smea_dy)
        setattr(k_diffusion_sampling, "sample_euler_negative", sample_euler_negative)
        setattr(k_diffusion_sampling, "sample_euler_dy_negative", sample_euler_dy_negative)

        SAMPLER_NAMES.append("euler_dy")
        SAMPLER_NAMES.append("euler_smea_dy")
        SAMPLER_NAMES.append("euler_negative")
        SAMPLER_NAMES.append("euler_dy_negative")

        smea_sampling.INITIALIZED = True

NODE_CLASS_MAPPINGS = {}