fix(oso_dagster): hotfix polars io manager with bq (#4672)

Author: ravenac95

pyproject.toml

@@ -67,6 +67,7 @@ dependencies = [
     "gcloud-aio-storage<10.0.0,>=9.3.0",
     "kr8s==0.20.9",
     "structlog>=25.4.0",
+    "pandas-gbq>=0.29.2",
 ]
 name = "oso"
 version = "1.0.0"

uv.lock

@@ -1,6 +1,6 @@
-import logging
 import typing as t
 
+import structlog
 from dagster import ConfigurableIOManagerFactory
 from dagster_dlt import DagsterDltResource
 from dagster_gcp import BigQueryResource, GCSResource
@@ -45,7 +45,7 @@
 from ..config import DagsterConfig
 from ..utils import GCPSecretResolver, LocalSecretResolver
 
-logger = logging.getLogger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 @resource_factory("global_config")

warehouse/oso_dagster/definitions/resources.py

@@ -1,9 +1,15 @@
+import structlog
 from dagster_duckdb_polars import DuckDBPolarsIOManager
 from dagster_polars import PolarsBigQueryIOManager
 from oso_dagster.config import DagsterConfig
 
+logger = structlog.get_logger(__name__)
+
 
 def load_io_manager(global_config: DagsterConfig):
     if global_config.enable_bigquery:
+        logger.info("Using PolarsBigQueryIOManager")
         return PolarsBigQueryIOManager(project=global_config.project_id)
+
+    logger.info("Using DuckDBPolarsIOManager")
     return DuckDBPolarsIOManager(database=global_config.local_duckdb_path)