feat: anchor run context to data package dir

z3z1ma · Nov 17, 2024 · c7866b3 · c7866b3
1 parent 05e6bd9
commit c7866b3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/cdf/core/extract_load.py b/src/cdf/core/extract_load.py
@@ -111,15 +111,19 @@ def _inject_provider(self):
         from dlt.common.configuration.container import Container
         from dlt.common.configuration.providers import CustomLoaderDocProvider
         from dlt.common.configuration.specs import PluggableRunContext
+        from dlt.common.runtime.run_context import RunContext
 
-        with Container().injectable_context(PluggableRunContext()) as dlt_context:
+        with Container().injectable_context(
+            PluggableRunContext(RunContext(run_dir=self.package_path))
+        ) as dlt_context:
             provider_name = f"cdf.{self.package_path.name}.configuration"
             if provider_name not in dlt_context.providers:
                 logger.debug("Injecting CDF configuration provider: %s", provider_name)
                 dlt_context.providers.add_provider(
                     CustomLoaderDocProvider(provider_name, lambda: self.config)
                 )
             yield
+            logger.debug("Restoring DLT context")
 
 
 class SlingPipelineProtocol(t.Protocol):