add get_sample_hashes() to NullFeatureExtractor

This commit is contained in:
Yacine Elhamer
2023-07-21 14:54:54 +01:00
parent 830bad54bd
commit 3d1a1fb9fa
+6
View File
@@ -62,6 +62,9 @@ class NullStaticFeatureExtractor(StaticFeatureExtractor):
for feature in self.global_features:
yield feature, NO_ADDRESS
def get_sample_hashes(self) -> SampleHashes:
return self.sample_hashes
def extract_file_features(self):
for address, feature in self.file_features:
yield feature, address
@@ -114,6 +117,9 @@ class NullDynamicFeatureExtractor(DynamicFeatureExtractor):
for feature in self.global_features:
yield feature, NO_ADDRESS
def get_sample_hashes(self) -> SampleHashes:
return self.sample_hashes
def extract_file_features(self):
for address, feature in self.file_features:
yield feature, address