Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
afoucret committed Apr 8, 2024
1 parent 5f41132 commit 3da831c
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 12 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -353,8 +353,11 @@ protected void ensureClusterStateConsistency() throws IOException {
);
entries.add(new NamedWriteableRegistry.Entry(Metadata.Custom.class, ModelAliasMetadata.NAME, ModelAliasMetadata::new));
entries.add(new NamedWriteableRegistry.Entry(NamedDiff.class, ModelAliasMetadata.NAME, ModelAliasMetadata::readDiffFrom));
entries.add(new NamedWriteableRegistry.Entry(Metadata.Custom.class, TrainedModelCacheMetadata.NAME, TrainedModelCacheMetadata::new));
entries.add(new NamedWriteableRegistry.Entry(NamedDiff.class, TrainedModelCacheMetadata.NAME, TrainedModelCacheMetadata::readDiffFrom));
entries.add(
new NamedWriteableRegistry.Entry(Metadata.Custom.class, TrainedModelCacheMetadata.NAME, TrainedModelCacheMetadata::new));
entries.add(
new NamedWriteableRegistry.Entry(NamedDiff.class, TrainedModelCacheMetadata.NAME, TrainedModelCacheMetadata::readDiffFrom)
);
entries.add(new NamedWriteableRegistry.Entry(Metadata.Custom.class, "ml", MlMetadata::new));
entries.add(new NamedWriteableRegistry.Entry(Metadata.Custom.class, IndexLifecycleMetadata.TYPE, IndexLifecycleMetadata::new));
entries.add(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1135,11 +1135,7 @@ public Collection<?> createComponents(PluginServices services) {
threadPool
);
final TrainedModelCacheManager trainedModelCacheManager = new TrainedModelCacheManager(clusterService);
final TrainedModelProvider trainedModelProvider = new TrainedModelProvider(
client,
trainedModelCacheManager,
xContentRegistry
);
final TrainedModelProvider trainedModelProvider = new TrainedModelProvider(client, trainedModelCacheManager, xContentRegistry);
final ModelLoadingService modelLoadingService = new ModelLoadingService(
trainedModelProvider,
inferenceAuditor,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -129,11 +129,7 @@ public class TrainedModelProvider {
private final NamedXContentRegistry xContentRegistry;
private final TrainedModelCacheManager modelCacheMetadataService;

public TrainedModelProvider(
Client client,
TrainedModelCacheManager modelCacheMetadataService,
NamedXContentRegistry xContentRegistry
) {
public TrainedModelProvider(Client client, TrainedModelCacheManager modelCacheMetadataService, NamedXContentRegistry xContentRegistry) {
this.client = client;
this.modelCacheMetadataService = modelCacheMetadataService;
this.xContentRegistry = xContentRegistry;
Expand Down

0 comments on commit 3da831c

Please sign in to comment.