Skip to content

Commit

Permalink
Use sql.bigquery.no_io for reference task
Browse files Browse the repository at this point in the history
Signed-off-by: Thomas J. Fan <thomasjpfan@gmail.com>
  • Loading branch information
thomasjpfan committed Aug 22, 2024
1 parent 361db73 commit 07f31a5
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions examples/productionizing/productionizing/reference_task.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from typing import List, NamedTuple
from typing import List

from flytekit import StructuredDataset, reference_task, workflow
from flytekit import reference_task, workflow
from flytekit.types.file import FlyteFile

# A `flytekit.reference_task` references the Flyte tasks that have already been defined, serialized, and registered.
Expand All @@ -26,16 +26,13 @@ def normalize_columns(
...


outputs = NamedTuple("Outputs", results=StructuredDataset)


@reference_task(
project="flytesnacks",
domain="development",
name="bigquery",
name="sql.bigquery.no_io",
version="{{ registration.version }}",
)
def bigquery_task(version: int) -> outputs:
def bigquery_task():
...


Expand Down

0 comments on commit 07f31a5

Please sign in to comment.