- modernize most of the postgis example. would like to do [ticket:1534] also.

- we don't have coverage for type-wide instrumentation events, the listener was broke.
could break again too.

1 files changed, 4 insertions, 2 deletions

diff --git a/lib/sqlalchemy/orm/events.py b/lib/sqlalchemy/orm/events.py
index 05f640ab0..25211d146 100644
--- a/lib/sqlalchemy/orm/events.py
+++ b/lib/sqlalchemy/orm/events.py
@@ -25,8 +25,9 @@ class InstrumentationEvents(event.Events):
 
     @classmethod
     def _accept_with(cls, target):
+        # TODO: there's no coverage for this
         if isinstance(target, type):
-            return orm.instrumentation.instrumentation_registry
+            return orm.instrumentation._instrumentation_factory
         else:
             return None
 
@@ -36,7 +37,8 @@ class InstrumentationEvents(event.Events):
 
     @classmethod
     def _remove(cls, identifier, target, fn):
-        raise NotImplementedError("Removal of instrumentation events not yet implemented")
+        raise NotImplementedError("Removal of instrumentation events "
+                                    "not yet implemented")
 
     def class_instrument(self, cls):
         """Called after the given class is instrumented.