ok

lexiforest · May 3, 2024 · d8e5e7c · d8e5e7c
1 parent 61f63f1
commit d8e5e7c
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/curl_cffi/requests/session.py b/curl_cffi/requests/session.py
@@ -83,13 +83,15 @@ def has(cls, item):
         return item in cls.__members__
 
     @classmethod
-    def normalize(cls, item) -> 'BrowserType':
-        browser_map = {
-            "chrome": cls.chrome,
-            "safari": cls.safari,
-            "safari_ios": cls.safari_ios,
-        }
-        return browser_map.get(item, item)
+    def normalize(cls, item):
+        if item == "chrome":  # noqa: SIM116
+            return cls.chrome
+        elif item == "safari":
+            return cls.safari
+        elif item == "safari_ios":
+            return cls.safari_ios
+        else:
+            return item
 
 class BrowserSpec:
     """A more structured way of selecting browsers"""