From affcca954963703c1611ba5a15f1f5117e61bedf Mon Sep 17 00:00:00 2001 From: yuhan Date: Wed, 14 Dec 2022 14:01:14 -0800 Subject: [PATCH] update AVAILABLE_EXAMPLES, isort, black --- .../quickstart_gcp/quickstart_gcp/assets/hackernews.py | 3 ++- examples/quickstart_gcp/quickstart_gcp/io_managers.py | 5 +++-- examples/quickstart_gcp/quickstart_gcp/repository.py | 10 +++++++--- python_modules/dagster/dagster/_generate/download.py | 1 + 4 files changed, 13 insertions(+), 6 deletions(-) diff --git a/examples/quickstart_gcp/quickstart_gcp/assets/hackernews.py b/examples/quickstart_gcp/quickstart_gcp/assets/hackernews.py index 60edcfde4250f..e305e071366e9 100644 --- a/examples/quickstart_gcp/quickstart_gcp/assets/hackernews.py +++ b/examples/quickstart_gcp/quickstart_gcp/assets/hackernews.py @@ -4,9 +4,10 @@ import matplotlib.pyplot as plt import pandas as pd import requests -from dagster import MetadataValue, OpExecutionContext, asset from wordcloud import STOPWORDS, WordCloud +from dagster import MetadataValue, OpExecutionContext, asset + @asset(group_name="hackernews", compute_kind="HackerNews API") def hackernews_topstory_ids() -> pd.DataFrame: diff --git a/examples/quickstart_gcp/quickstart_gcp/io_managers.py b/examples/quickstart_gcp/quickstart_gcp/io_managers.py index c8cd40910e05b..96007d598f7e2 100644 --- a/examples/quickstart_gcp/quickstart_gcp/io_managers.py +++ b/examples/quickstart_gcp/quickstart_gcp/io_managers.py @@ -2,16 +2,17 @@ import pandas as pd import pandas_gbq +from google.oauth2 import service_account + from dagster import ( Field, + IOManager, InitResourceContext, InputContext, - IOManager, OutputContext, StringSource, io_manager, ) -from google.oauth2 import service_account # Learn more about custom I/O managers in Dagster docs: # https://docs.dagster.io/concepts/io-management/io-managers#a-custom-io-manager-that-stores-pandas-dataframes-in-tables diff --git a/examples/quickstart_gcp/quickstart_gcp/repository.py b/examples/quickstart_gcp/quickstart_gcp/repository.py index 0b19ed0f4209e..7b169c7ded7f5 100644 --- a/examples/quickstart_gcp/quickstart_gcp/repository.py +++ b/examples/quickstart_gcp/quickstart_gcp/repository.py @@ -1,6 +1,10 @@ -from dagster import (ScheduleDefinition, define_asset_job, - load_assets_from_package_module, repository, - with_resources) +from dagster import ( + ScheduleDefinition, + define_asset_job, + load_assets_from_package_module, + repository, + with_resources, +) from . import assets from .io_managers import bigquery_pandas_io_manager diff --git a/python_modules/dagster/dagster/_generate/download.py b/python_modules/dagster/dagster/_generate/download.py index 392b42f717a51..8e1864189ad7c 100644 --- a/python_modules/dagster/dagster/_generate/download.py +++ b/python_modules/dagster/dagster/_generate/download.py @@ -22,6 +22,7 @@ "assets_smoke_test", "quickstart_aws", "quickstart_etl", + "quickstart_gcp", "tutorial_dbt_dagster", "tutorial_notebook_assets", "deploy_docker",