Files
lm-dataset-preparer/src/main/java/de/extio/lmdatasetprep/tools/DatasetTool.java
2024-12-20 00:14:02 +01:00

25 lines
700 B
Java

package de.extio.lmdatasetprep.tools;
import java.util.Properties;
import java.util.function.Consumer;
import de.extio.lmlib.client.Client;
import de.extio.lmlib.client.ClientService;
import de.extio.lmlib.profile.ModelCategory;
public interface DatasetTool extends Consumer<Properties> {
default String getModelCategoryPropertyName() {
return null;
}
default ModelCategory getModelCategory(final Properties properties) {
return ModelCategory.valueOf(properties.getProperty(this.getModelCategoryPropertyName()));
}
default Client getClient(final Properties properties, final ClientService clientService) {
return clientService.getClient(this.getModelCategory(properties));
}
}