Move the rest of pokedex.util.get to pokedex.db.util

[zzz-pokedex.git] / pokedex / db / load.py

diff --git a/pokedex/db/load.py b/pokedex/db/load.py
index 5740f42..d307fd8 100644 (file)
--- a/pokedex/db/load.py
+++ b/pokedex/db/load.py
@@ -295,7 +295,7 @@ def load(session, tables=[], directory=None, drop_tables=False, verbose=False, s
             session.connection().execute(
                 insert_stmt.values(**row_data)
             )
-            seen_ids[row_data['id']] = 1
+            seen_ids.add(row_data['id'])
         session.commit()
 
         print_done()