diff --git a/python_modules/libraries/dagster-polars/dagster_polars/io_managers/__init__.py b/python_modules/libraries/dagster-polars/dagster_polars/io_managers/__init__.py index 7e9c959a4bd53..1df29edf4ccac 100644 --- a/python_modules/libraries/dagster-polars/dagster_polars/io_managers/__init__.py +++ b/python_modules/libraries/dagster-polars/dagster_polars/io_managers/__init__.py @@ -28,7 +28,7 @@ __all__.extend(["PolarsBigQueryIOManager", "PolarsBigQueryTypeHandler"]) except ImportError as e: - if "google-cloud-bigquery" in str(e): + if "google-cloud-bigquery" in str(e) or "dagster-polars[gcp]" in str(e): pass else: raise e