Merge pull request carykh#13 from t1m0thyj/patch-2

Make list of strategy folders dynamic
valadaptive · May 25, 2021 · 266f939 · 266f939
2 parents 861f4f5 + c7797ce
commit 266f939
Showing 1 changed file with 3 additions and 18 deletions.
diff --git a/code/prisonersDilemma.py b/code/prisonersDilemma.py
@@ -96,24 +96,9 @@
 
 args = parser.parse_args()
 
-STRATEGY_FOLDERS = [
-    "exampleStrats",
-    "valadaptive",
-    "nekiwo",
-    "edward",
-    "misc",
-    "saffron",
-    "aaaa-trsh",
-    "phoenix",
-    "l4vr0v",
-    "smough",
-    "dratini0",
-    "decxjo",
-    "Nobody5050",
-    "sanscipher"
-]
-if args.use_slow:
-    STRATEGY_FOLDERS.append("slow")
+STRATEGY_FOLDERS = [p for p in os.listdir() if os.path.isdir(p)]
+if not args.use_slow:
+    STRATEGY_FOLDERS.remove("slow")
 RESULTS_FILE = "results.txt"
 RESULTS_HTML = "results.html"
 RESULTS_JSON = "results.json"