Skip to main content
Version: v3.0.0

Snowflake example

import h2o_featurestore
from h2o_featurestore import SnowflakeTable, Proxy

# Initialise feature store client
client = h2o_featurestore.login()

# Set workspace specifics
workspace = client.workspaces.create("demo")

# Create a snowflake table source
proxy = Proxy(host="proxy host", port=8080, user="username", password="password")
snowflake_table = SnowflakeTable(url="<snowflake account url>", warehouse="warehouse name", database="database name", schema="schema name", table="table name", insecure=False, proxy=proxy, role="role", account="account name")
snowflake_table_schema = workspace.extract_schema_from_source(snowflake_table)

# Create a snowflake query source
snowflake_query = SnowflakeTable(url="<snowflake account url>", warehouse="warehouse name", database="database name", schema="schema name", query="<sql select query>")
snowflake_query_schema = workspace.extract_schema_from_source(snowflake_query)

# Register the feature set
my_feature_set_1 = workspace.feature_sets.register(snowflake_table_schema, "feature_set_1", primary_key=["key_name"])
my_feature_set_2 = workspace.feature_sets.register(snowflake_query_schema, "feature_set_2", primary_key=["key_name"])

# Ingest to cache
my_feature_set_1.ingest(snowflake_table)
my_feature_set_2.ingest(snowflake_query)

# Retrieve feature set
ref = my_feature_set_1.retrieve()
ref.download()
ref = my_feature_set_2.retrieve()
ref.download()
Feedback