scripts/profile_time.py: revert restriction that frozen extractors can only be static ones

This commit is contained in:
Yacine Elhamer
2023-10-04 10:51:02 +02:00
parent 35f64f37bb
commit 28a722d4c3

View File

@@ -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