From 28a722d4c338f6372550bf418902e240c1bb302d Mon Sep 17 00:00:00 2001 From: Yacine Elhamer Date: Wed, 4 Oct 2023 10:51:02 +0200 Subject: [PATCH] scripts/profile_time.py: revert restriction that frozen extractors can only be static ones --- scripts/profile-time.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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