From bb34d4e13da0f4753fbdefee837f54b16149d3ef Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Thu, 30 Aug 2018 00:45:21 +0000
Subject: [PATCH] Skeleton adjustment, partial implementiation of image display for generator

---
 generate_data.py |    6 ++----
 1 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/generate_data.py b/generate_data.py
index d3e7d4e..54f4d4d 100644
--- a/generate_data.py
+++ b/generate_data.py
@@ -181,15 +181,13 @@
 
 
 def main():
+    random.seed()
     #bg_images = load_dtd()
     #bg = Backgrounds()
     #bg.get_random(display=True)
 
     card_pool = pd.DataFrame()
-    for set_name in ['8ed', 'mrd', 'dst', '5dn', 'chk', 'bok', 'sok', '9ed', 'rav', 'gpt', 'dis', 'csp', 'tsp', 'plc',
-                     'fut', '10e', 'lrw', 'mor', 'shm', 'eve', 'ala', 'con', 'arb', 'm10', 'zen', 'wwk', 'roe', 'm11',
-                     'som', 'mbs', 'nph', 'm12', 'isd', 'dka', 'avr', 'm13', 'rtr', 'gtc', 'dgm', 'm14', 'ths', 'bng',
-                     'jou']:
+    for set_name in fetch_data.all_set_list:
         df = fetch_data.load_all_cards_text('data/csv/%s.csv' % set_name)
         for _ in range(3):
             card_info = df.iloc[random.randint(0, df.shape[0] - 1)]

--
Gitblit v1.10.0