add initial value

PrefectHQ · Apr 24, 2024 · 338e770 · 338e770
1 parent 142082b
commit 338e770
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/marvin/utilities/context.py b/src/marvin/utilities/context.py
@@ -26,10 +26,10 @@ class ScopedContext:
         ```
     """
 
-    def __init__(self):
-        """Initializes the ScopedContext with a default empty dictionary."""
+    def __init__(self, initial_value: dict = None):
+        """Initializes the ScopedContext with an initial valuedictionary."""
         self._context_storage = contextvars.ContextVar(
-            "scoped_context_storage", default={}
+            "scoped_context_storage", default=initial_value or {}
         )
 
     def get(self, key: str, default: Any = None) -> Any: