Skip to content

Commit

Permalink
fix imports
Browse files Browse the repository at this point in the history
  • Loading branch information
mikealfare committed Oct 31, 2024
1 parent cc0803d commit 4dfa6b1
Show file tree
Hide file tree
Showing 6 changed files with 21 additions and 17 deletions.
5 changes: 3 additions & 2 deletions dbt/adapters/bigquery/column.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
from dataclasses import dataclass
from typing import Optional, List, TypeVar, Iterable, Type, Any, Dict, Union
from typing import Any, Dict, Iterable, List, Optional, Type, TypeVar, Union

from google.cloud.bigquery import SchemaField

from dbt.adapters.base.column import Column

from google.cloud.bigquery import SchemaField

_PARENT_DATA_TYPE_KEY = "__parent_data_type"

Expand Down
2 changes: 1 addition & 1 deletion dbt/adapters/bigquery/connections.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
from dbt.adapters.events.types import SQLQuery
from dbt.adapters.exceptions.connection import FailedToConnectError

from dbt.adapters.bigquery import __version__ as dbt_version
import dbt.adapters.bigquery.__version__ as dbt_version
from dbt.adapters.bigquery import gcloud
from dbt.adapters.bigquery.credentials import (
BigQueryConnectionMethod,
Expand Down
4 changes: 3 additions & 1 deletion dbt/adapters/bigquery/dataset.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
from typing import List
from google.cloud.bigquery import Dataset, AccessEntry

from google.cloud.bigquery import AccessEntry, Dataset

from dbt.adapters.events.logging import AdapterLogger


logger = AdapterLogger("BigQuery")


Expand Down
4 changes: 2 additions & 2 deletions dbt/adapters/bigquery/gcloud.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from dbt_common.clients.system import run_cmd
from dbt_common.exceptions import DbtRuntimeError

from dbt.adapters.events.logging import AdapterLogger
from dbt_common.clients.system import run_cmd


NOT_INSTALLED_MSG = """
dbt requires the gcloud SDK to be installed to authenticate with BigQuery.
Expand Down
14 changes: 7 additions & 7 deletions dbt/adapters/bigquery/python_submissions.py
Original file line number Diff line number Diff line change
@@ -1,21 +1,21 @@
import uuid
from typing import Dict, Union

from dbt.adapters.events.logging import AdapterLogger

from dbt.adapters.base import PythonJobHelper
from google.api_core.future.polling import POLLING_PREDICATE

from dbt.adapters.bigquery import BigQueryConnectionManager, BigQueryCredentials
from google.api_core import retry
from google.api_core.client_options import ClientOptions
from google.api_core.future.polling import POLLING_PREDICATE
from google.cloud import storage, dataproc_v1
from google.cloud.dataproc_v1.types.batches import Batch

from dbt.adapters.base import PythonJobHelper
from dbt.adapters.events.logging import AdapterLogger

from dbt.adapters.bigquery.connections import BigQueryConnectionManager
from dbt.adapters.bigquery.credentials import BigQueryCredentials
from dbt.adapters.bigquery.dataproc.batch import (
DEFAULT_JAR_FILE_URI,
create_batch_request,
poll_batch_job,
DEFAULT_JAR_FILE_URI,
update_batch_from_config,
)

Expand Down
9 changes: 5 additions & 4 deletions dbt/adapters/bigquery/relation.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,20 @@
from dataclasses import dataclass, field
from itertools import chain, islice
from typing import FrozenSet, Optional, TypeVar

from itertools import chain, islice
from dbt_common.exceptions import CompilationError
from dbt_common.utils.dict import filter_null_values
from dbt.adapters.base.relation import BaseRelation, ComponentName, InformationSchema
from dbt.adapters.contracts.relation import RelationConfig, RelationType
from dbt.adapters.relation_configs import RelationConfigChangeAction

from dbt.adapters.bigquery.relation_configs import (
BigQueryClusterConfigChange,
BigQueryMaterializedViewConfig,
BigQueryMaterializedViewConfigChangeset,
BigQueryOptionsConfigChange,
BigQueryPartitionConfigChange,
)
from dbt.adapters.contracts.relation import RelationType, RelationConfig
from dbt_common.exceptions import CompilationError
from dbt_common.utils.dict import filter_null_values


Self = TypeVar("Self", bound="BigQueryRelation")
Expand Down

0 comments on commit 4dfa6b1

Please sign in to comment.