diff --git a/python/kvikio/kvikio/defaults.py b/python/kvikio/kvikio/defaults.py
index 0b2d396b3f..181382cdcf 100644
--- a/python/kvikio/kvikio/defaults.py
+++ b/python/kvikio/kvikio/defaults.py
@@ -36,6 +36,16 @@ def _set_property(self, property: str, value: Any):
 
 
 def set(*config: Any) -> ConfigContextManager:
+    """Set KvikIO configurations.
+
+    Parameters
+    -------
+    Any
+        The configurations. Can either be a single parameter (dict) consisting of one
+        or more properties, or two parameters key (string) and value (Any)
+        indicating a single property.
+    """
+
     err_msg = (
         "Valid arguments are kvikio.defaults.set(config: dict) or "
         "kvikio.defaults.set(key: str, value: Any)"