Add more build files to extractor based on analysis

linkvt · Jan 4, 2017 · d7cbc37 · d7cbc37
1 parent 57b6e82
commit d7cbc37
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/classification/feature_extraction/dev.py b/classification/feature_extraction/dev.py
@@ -8,7 +8,9 @@
 
 class HasBuildFileExtractor(FeatureExtractor):
     # TODO: Add more build files
-    build_files = ['build.gradle', 'composer.json', 'makefile' 'package.json', 'pom.xml']
+    build_files = ['.travis.yml', 'appveyor.yml', 'circle.yml', 'CMakeLists.txt', 'cmake', 'tox.ini', 'setup.py',
+                   'build.gradle', 'gradle', 'gradlew', 'composer.json', 'makefile' 'package.json', 'pom.xml',
+                   'requirements.txt', 'manifest.json', 'Package.swift']
 
     def _init_features(self):
         self.features = [Feature('Has build file')]