diff --git a/scripts/profile-time.py b/scripts/profile-time.py index cb2d9368..cd4283f7 100644 --- a/scripts/profile-time.py +++ b/scripts/profile-time.py @@ -105,8 +105,7 @@ def main(argv=None): if (args.format == "freeze") or ( args.format == capa.features.common.FORMAT_AUTO and capa.features.freeze.is_freeze(taste) ): - extractor: FeatureExtractor = capa.features.freeze.load(Path(args.sample).read_bytes()) - assert isinstance(extractor, StaticFeatureExtractor) + extractor = capa.features.freeze.load(Path(args.sample).read_bytes()) else: extractor = capa.main.get_extractor( args.sample, args.format, args.os, capa.main.BACKEND_VIV, sig_paths, should_save_workspace=False