|  | @@ -28,8 +28,8 @@ import org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceConfig;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.inference.trainedmodel.LenientlyParsedInferenceConfig;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.inference.trainedmodel.StrictlyParsedInferenceConfig;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.inference.trainedmodel.metadata.FeatureImportanceBaseline;
 | 
	
		
			
				|  |  | -import org.elasticsearch.xpack.core.ml.inference.trainedmodel.metadata.TotalFeatureImportance;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.inference.trainedmodel.metadata.Hyperparameters;
 | 
	
		
			
				|  |  | +import org.elasticsearch.xpack.core.ml.inference.trainedmodel.metadata.TotalFeatureImportance;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.job.messages.Messages;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.utils.ExceptionsHelper;
 | 
	
		
			
				|  |  |  import org.elasticsearch.xpack.core.ml.utils.MlStrings;
 | 
	
	
		
			
				|  | @@ -40,11 +40,9 @@ import java.time.Instant;
 | 
	
		
			
				|  |  |  import java.util.Arrays;
 | 
	
		
			
				|  |  |  import java.util.Collections;
 | 
	
		
			
				|  |  |  import java.util.HashMap;
 | 
	
		
			
				|  |  | -import java.util.HashSet;
 | 
	
		
			
				|  |  |  import java.util.List;
 | 
	
		
			
				|  |  |  import java.util.Map;
 | 
	
		
			
				|  |  |  import java.util.Objects;
 | 
	
		
			
				|  |  | -import java.util.Set;
 | 
	
		
			
				|  |  |  import java.util.stream.Collectors;
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  import static org.elasticsearch.action.ValidateActions.addValidationError;
 | 
	
	
		
			
				|  | @@ -60,9 +58,6 @@ public class TrainedModelConfig implements ToXContentObject, Writeable {
 | 
	
		
			
				|  |  |      public static final String TOTAL_FEATURE_IMPORTANCE = "total_feature_importance";
 | 
	
		
			
				|  |  |      public static final String FEATURE_IMPORTANCE_BASELINE = "feature_importance_baseline";
 | 
	
		
			
				|  |  |      public static final String HYPERPARAMETERS = "hyperparameters";
 | 
	
		
			
				|  |  | -    private static final Set<String> RESERVED_METADATA_FIELDS = new HashSet<>(Arrays.asList(
 | 
	
		
			
				|  |  | -        TOTAL_FEATURE_IMPORTANCE,
 | 
	
		
			
				|  |  | -        FEATURE_IMPORTANCE_BASELINE));
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      private static final String ESTIMATED_HEAP_MEMORY_USAGE_HUMAN = "estimated_heap_memory_usage";
 | 
	
		
			
				|  |  |  
 |