Fixed leaky connection

posthog/temporal/data_imports/pipelines/rest_source/__init__.py

@@ -372,22 +372,3 @@ def identity_func(x: Any) -> Any:
     if incremental_param.end:
         params[incremental_param.end] = transform(incremental_object.end_value)
     return params
-
-
-# XXX: This is a workaround pass test_dlt_init.py
-# since the source uses dlt.source as a function
-# def _register_source(source_func: Callable[..., DltSource]) -> None:
-#     import inspect
-#     from dlt.common.configuration import get_fun_spec
-#     from dlt.common.source import _SOURCES, SourceInfo
-
-#     spec = get_fun_spec(source_func)
-#     func_module = inspect.getmodule(source_func)
-#     _SOURCES[source_func.__name__] = SourceInfo(
-#         SPEC=spec,
-#         f=source_func,
-#         module=func_module,
-#     )
-
-
-# _register_source(rest_api_source)

posthog/temporal/data_imports/pipelines/sql_database_v2/__init__.py

@@ -220,10 +220,7 @@ def sql_database(
 
     # use provided tables or all tables
     if table_names:
-        tables = [
-            Table(name, metadata, inherit_cache=False, autoload_with=None if defer_table_reflect else engine)
-            for name in table_names
-        ]
+        tables = [Table(name, metadata, autoload_with=None if defer_table_reflect else engine) for name in table_names]
     else:
         if defer_table_reflect:
             raise ValueError("You must pass table names to defer table reflection")
@@ -232,7 +229,7 @@ def sql_database(
 
     for table in tables:
         yield sql_table(
-            credentials=credentials,
+            credentials=engine,
             table=table.name,
             schema=table.schema,
             metadata=metadata,