ContrastiveSampler lacks a seed param, while the docstring states it has one
Closed this issue · 1 comments
martin-gorner commented
Contrastive samples docs here:
https://keras.io/api/keras_nlp/samplers/contrastive_sampler/
The function signature does not have a "seed" arg:
keras_nlp.samplers.ContrastiveSampler(k=5, alpha=0.6, **kwargs)
But the docstring states:
seed: int. The random seed. Defaults to None.
Probably just a docstring issue. Since this samples is a variant of BeamSampler, it probably does not need a seed.
mattdangerw commented
Yep, just a docs issue. Will fix up.