Make pokedex work with SQLAlchemy 0.7. Warning: ugly hack!

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

diff --git a/pokedex/db/load.py b/pokedex/db/load.py
index 5740f42..d0dc00b 100644 (file)
--- a/pokedex/db/load.py
+++ b/pokedex/db/load.py
@@ -4,7 +4,6 @@ import fnmatch
 import os.path
 import sys
 
-from sqlalchemy.orm.attributes import instrumentation_registry
 import sqlalchemy.sql.util
 import sqlalchemy.types
 
@@ -295,7 +294,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()