pep8

capa/features/extractors/base_extractor.py

@@ -58,7 +58,7 @@ class FeatureExtractor:
         fetch the preferred load address at which the sample was analyzed.
         """
         raise NotImplemented
- 
+
     @abc.abstractmethod
     def extract_global_features(self) -> Iterator[Tuple[Feature, int]]:
         """
@@ -74,7 +74,7 @@ class FeatureExtractor:
           Tuple[Feature, int]: feature and its location
         """
         raise NotImplemented
- 
+
     @abc.abstractmethod
     def extract_file_features(self) -> Iterator[Tuple[Feature, int]]:
         """
@@ -272,12 +272,12 @@ class NullFeatureExtractor(FeatureExtractor):
 
     def get_base_address(self):
         return self.features["base address"]
-                                             
+
     def extract_global_features(self):
         for p in self.features.get("global features", []):
             va, feature = p
             yield feature, va
-                                             
+
     def extract_file_features(self):
         for p in self.features.get("file features", []):
             va, feature = p

capa/features/extractors/pefile.py

@@ -150,7 +150,7 @@ FILE_HANDLERS = (
     extract_file_function_names,
     extract_file_format,
 )
- 
+
 
 def extract_global_features(pe, buf):
     """

capa/features/freeze.py

@@ -192,7 +192,7 @@ def loads(s):
         va = int(va, 0x10)
         feature = deserialize_feature(feature[:2])
         features["global features"].append((va, feature))
- 
+
     for feature in doc.get("scopes", {}).get("file", []):
         va, loc = feature[2:]
         va = int(va, 0x10)

capa/main.py

@@ -88,12 +88,16 @@ def find_function_capabilities(ruleset: RuleSet, extractor: FeatureExtractor, f:
         #  - basic blocks
         bb_features = collections.defaultdict(set)
 
-        for feature, va in itertools.chain(extractor.extract_basic_block_features(f, bb), extractor.extract_global_features()):
+        for feature, va in itertools.chain(
+            extractor.extract_basic_block_features(f, bb), extractor.extract_global_features()
+        ):
             bb_features[feature].add(va)
             function_features[feature].add(va)
 
         for insn in extractor.get_instructions(f, bb):
-            for feature, va in itertools.chain(extractor.extract_insn_features(f, bb, insn), extractor.extract_global_features()):
+            for feature, va in itertools.chain(
+                extractor.extract_insn_features(f, bb, insn), extractor.extract_global_features()
+            ):
                 bb_features[feature].add(va)
                 function_features[feature].add(va)
 

tests/fixtures.py

@@ -139,7 +139,7 @@ def extract_global_features(extractor):
     for feature, va in extractor.extract_global_features():
         features[feature].add(va)
     return features
- 
+
 
 @lru_cache()
 def extract_file_features(extractor):