Merge pull request #4 from r0mainK/fix/value

Fix value in path extractor

Author: zurk

src/algorithms/uast_to_bag_paths.py

@@ -27,7 +27,7 @@ def __call__(self, uast):
         """
 
         path_contexts = get_paths(uast, self._max_length, self._max_width)
-        dict_of_paths = Counter(path_contexts)
+        dict_of_paths = {str(path): val for path, val in Counter(path_contexts).items()}
         self._log.info("Extracted paths successfully")
 
         from pprint import pprint

src/extractors/paths.py

@@ -12,4 +12,4 @@ def __init__(self, max_length=5, max_width=5, **kwargs):
         self.uast2paths = Uast2BagOfPaths(max_length, max_width)
 
     def uast_to_bag(self, uast):
-        return {str(path): 1 for path in self.uast2paths(uast)}
+        return self.uast2paths(uast)