Skip to content

Commit fa5759e

Browse files
committed
added duckdb workflow postgres
1 parent f1997a5 commit fa5759e

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

hts_dagster_sample/assets/postgres_workflow.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ def generate_users():
1313
return df
1414

1515

16-
@asset
17-
def update_user_database(generate_users, postgres_db):
16+
@asset(required_resource_keys={"postgres"}) # Declare postgres as required resource
17+
def update_user_database(context, generate_users): # Add context, remove postgres_db parameter
1818
"""Update PostgreSQL database with user data"""
19-
with postgres_db.get_connection() as conn:
19+
with context.resources.postgres.get_connection() as conn: # Use postgres through context
2020
# Create table if not exists
2121
create_table_sql = """
2222
CREATE TABLE IF NOT EXISTS users (

0 commit comments

Comments
 (0)