patched SmaryQuery._get_root_cls

absent1706 · Jan 28, 2024 · 6da12c5 · 6da12c5
1 parent 285a530
commit 6da12c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sqlalchemy_mixins/activerecordasync.py b/sqlalchemy_mixins/activerecordasync.py
@@ -5,17 +5,17 @@
 from .activerecord import ModelNotFoundError
 from . import smartquery as SmaryQuery
 
-
+get_root_cls = SmaryQuery._get_root_cls
 def async_root_cls(query):
     """Monkey patch SmaryQuery to handle async queries."""
     try:
-        return SmaryQuery._get_root_cls(query)
+        return get_root_cls(query)
     except ValueError:
         # Handle async queries
         if query.__dict__["_propagate_attrs"]["plugin_subject"].class_:
             return query.__dict__["_propagate_attrs"]["plugin_subject"].class_
         raise
-    
+
 SmaryQuery._get_root_cls = lambda query: async_root_cls(query)