def probabilities(self, probabilities: dict):
if probabilities is None:
self._probabilities = None
elif isinstance(probabilities, str):
self._probabilities = {validate_str(probabilities): 1.0}
elif isinstance(probabilities, dict):
if abs(sum(probabilities.values()) - 1.0) > 1.0e-9:
raise ValueError("probabilities must sum to 1.0")
self._probabilities = keymap(validate_str, probabilities)
else:
raise TypeError("probabilities must be dict or single value")
def name(self, name):
if name is None:
self._name = None
else:
self._name = validate_str(name)
def name(self, name: str):
self._name = validate_str(name)
def category(self, category):
if category is None:
self._category = None
else:
self._category = validate_str(category)