From 8169d8aacd8598730fd2e6eba06052e7502f2cc1 Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Mon, 1 Nov 2021 09:05:32 +0300
Subject: Fix mypy issues

---
 gn3/authentication.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'gn3')

diff --git a/gn3/authentication.py b/gn3/authentication.py
index 892aa8f..7bc7b77 100644
--- a/gn3/authentication.py
+++ b/gn3/authentication.py
@@ -3,9 +3,9 @@ import functools
 import json
 from urllib.parse import urljoin
 from enum import Enum, unique
-from typing import Dict
+from typing import Dict, Union
 
-import redis
+from redis import Redis
 import requests
 
 
@@ -40,7 +40,7 @@ class AdminRole(OrderedEnum):
     EDIT_ADMINS = "edit-admins"
 
 
-def get_user_membership(conn: redis.Redis, user_id: str,
+def get_user_membership(conn: Redis, user_id: str,
                         group_id: str) -> Dict:
     """Return a dictionary that indicates whether the `user_id` is a
     member or admin of `group_id`.
@@ -84,7 +84,7 @@ def get_highest_user_access_role(
       A dict indicating the highest access role the user has.
 
     """
-    role_mapping = {}
+    role_mapping: Dict[str, Union[DataRole, AdminRole]] = {}
     for data_role, admin_role in zip(DataRole, AdminRole):
         role_mapping.update({data_role.value: data_role, })
         role_mapping.update({admin_role.value: admin_role, })
-- 
cgit v1.2.3