From 2dde984e88e1b30144f35e11dc12824bf3e470aa Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Wed, 3 Oct 2012 11:10:42 -0400 Subject: - add one more #2583 test to cover the "multiple PK switch" use case --- lib/sqlalchemy/orm/session.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'lib/sqlalchemy') diff --git a/lib/sqlalchemy/orm/session.py b/lib/sqlalchemy/orm/session.py index faa9e5a83..1df9d45ca 100644 --- a/lib/sqlalchemy/orm/session.py +++ b/lib/sqlalchemy/orm/session.py @@ -1288,7 +1288,7 @@ class Session(_SessionClassMethods): # map (see test/orm/test_naturalpks.py ReversePKsTest) self.identity_map.discard(state) if state in self.transaction._key_switches: - orig_key = self.transaction._key_switches[0] + orig_key = self.transaction._key_switches[state][0] else: orig_key = state.key self.transaction._key_switches[state] = (orig_key, instance_key) -- cgit v1.2.1