Minor adjustment

rapidsai · Feb 25, 2025 · b7ac4ee · b7ac4ee
1 parent fd8542f
commit b7ac4ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/kvikio/kvikio/defaults.py b/python/kvikio/kvikio/defaults.py
@@ -37,7 +37,7 @@ def wrapper(*args, **kwargs):
 
 
 class ConfigContextManager:
-    def __init__(self, config: dict):
+    def __init__(self, config: dict[str, str]):
         (
             self._all_getter_property_functions,
             self._all_setter_property_functions,
@@ -51,7 +51,7 @@ def __init__(self, config: dict):
     def __enter__(self):
         return None
 
-    def __exit__(self, type, value, traceback):
+    def __exit__(self, type_unused, value, traceback_unused):
         for key, value in self._old_properties.items():
             self._set_property(key, value)