Fix code after incoming changes from opus-42

opus-42 · Aug 21, 2022 · 5cd9f32 · 5cd9f32
1 parent bbadbf1
commit 5cd9f32
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/supersetapiclient/client.py b/supersetapiclient/client.py
@@ -16,6 +16,7 @@
 from supersetapiclient.datasets import Datasets
 from supersetapiclient.databases import Databases
 from supersetapiclient.saved_queries import SavedQueries
+from supersetapiclient.roles import Roles
 from supersetapiclient.exceptions import QueryLimitReached
 
 logger = logging.getLogger(__name__)
@@ -28,6 +29,7 @@ class SupersetClient:
     datasets_cls = Datasets
     databases_cls = Databases
     saved_queries_cls = SavedQueries
+    roles_cls = Roles
     http_adapter_cls = None
 
     def __init__(
@@ -52,7 +54,7 @@ def __init__(
         self.datasets = self.datasets_cls(self)
         self.databases = self.databases_cls(self)
         self.saved_queries = self.saved_queries_cls(self)
-        self.roles = Roles(self)
+        self.roles = self.roles_cls(self)
 
     @cached_property
     def _token(self):