[ML][Training] Dataset.setProperty() refactoring