from .eval_sampler import EvalSampler, EvalSamplerAdvanced

NODE_CLASS_MAPPINGS = {
    "EvalSampler": EvalSampler,
    "EvalSamplerAdvanced": EvalSamplerAdvanced
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "EvalSampler": "Eval Sampler (Advanced)",
    "EvalSamplerAdvanced": "Eval Sampler (Custom Advanced)"
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]