refactor: Update deprecated class usage

orobix · Jun 3, 2024 · 001a1ef · 001a1ef
1 parent b3fbbcf
commit 001a1ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/anomalib/models/padim/torch_model.py b/src/anomalib/models/padim/torch_model.py
@@ -14,6 +14,7 @@
 
 from anomalib.models.components import FeatureExtractor, MultiVariateGaussian
 from anomalib.models.components.feature_extractors import dryrun_find_featuremap_dims
+from anomalib.models.components.feature_extractors.timm import TimmFeatureExtractor
 from anomalib.models.padim.anomaly_map import AnomalyMapGenerator
 from anomalib.pre_processing import Tiler
 
@@ -87,7 +88,7 @@ def __init__(
         self.layers = layers
         self.backbone = backbone
 
-        self.feature_extractor = FeatureExtractor(
+        self.feature_extractor = TimmFeatureExtractor(
             backbone=self.backbone, layers=layers, pre_trained=pre_trained, pretrained_weights=pretrained_weights
         )
         self.n_features_original, self.n_patches = _deduce_dims(self.feature_extractor, input_size, self.layers)