Merge pull request #189 from fuhrysteve/reraise_original_exception

Reraise original exception instead of a new one
kvesteri · Jun 12, 2018 · 5a3fbcf · 5a3fbcf
2 parents d4ed900 + 79b9876
commit 5a3fbcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sqlalchemy_continuum/manager.py b/sqlalchemy_continuum/manager.py
@@ -32,7 +32,7 @@ def wrapper(self, mapper, connection, target):
                         uow = self.unit_of_work(session)
                         break  # The ConnectionFairy is the same, this connection is a clone
                 else:
-                    raise KeyError
+                    raise
         return func(self, uow, target)
     return wrapper
 
@@ -400,7 +400,7 @@ def append_association_operation(self, conn, table_name, params, op):
                         uow = self.unit_of_work(conn.session)
                         break  # The ConnectionFairy is the same, this connection is a clone
                 else:
-                    raise KeyError
+                    raise
         uow.pending_statements.append(stmt)
 
     def track_cloned_connections(self, c, opt):