improved run.py

MarkusRabe · Mar 16, 2019 · 6151c48 · 6151c48
1 parent 2bcf0e0
commit 6151c48
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scripts/run.py b/scripts/run.py
@@ -80,10 +80,10 @@ def get_paths_from_categories(categories, directory):
             omitted_files = 0
             detected_files = 0
             for filename in filenames:
-                if filename.endswith('qdimacs.gz') or \
-                   filename.endswith('aag') or \
-                   filename.endswith('aig') or \
-                   filename.endswith('qdimacs'):
+                if filename.endswith('.qdimacs.gz') or \
+                   filename.endswith('.aag') or \
+                   filename.endswith('.aig') or \
+                   filename.endswith('.qdimacs'):
 
                     paths[os.path.join(dirpath,filename)] = (os.path.join(dirpath,filename), 30)
                     detected_files += 1