INNER CODE UNIT · Python
__post_init__
microsoft/muzic · musecoco/1-text2attribute_model/main.py:120
def __post_init__(self):
if (self.train_file or self.validation_file or self.test_file) is None:
raise ValueError("Need either a training/validation/test file.")
elif self.train_file and self.validation_file:
train_extension = self.train_file.split(".")[-1]
assert train_extension=="json", "`train_file` should be a json file."
validation_extension = self.validation_file.split(".")[-1]
assert (
validation_extension == train_extension
), "`validation_file` should have the same extension (json) as `train_file`."
if self.test_file:
test_extension = self.test_file.split(".")[-1]
assert (
test_extension=="json"
), "`test_file` should have the extension `json`."
if self.attributes is None: