diff --git a/thingsvision/core/extraction/extractors.py b/thingsvision/core/extraction/extractors.py index 7928825..f91728f 100644 --- a/thingsvision/core/extraction/extractors.py +++ b/thingsvision/core/extraction/extractors.py @@ -238,7 +238,7 @@ def get_keras_preprocessing(self, model_name:str) -> Union[str, None]: ] # Try each pattern for pattern, preproc_val in patterns: - if re.match(pattern, model_name): + if re.search(pattern, model_name): return preproc_val # If no match is found, print a warning message