fix the typo

Signed-off-by: HH <[email protected]>
flyteorg · Sep 17, 2023 · 13c6d5b · 13c6d5b
1 parent 96aae78
commit 13c6d5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flytekit/types/structured/snowflake.py b/flytekit/types/structured/snowflake.py
@@ -24,7 +24,7 @@ def get_private_key():
 
     import flytekit
 
-    pk_string = flytekit.current_context().secrets.get(TASK_TYPE, "private_key", encode_mode="rb")
+    pk_string = flytekit.current_context().secrets.get(SNOWFLAKE, "private_key", encode_mode="rb")
     p_key = serialization.load_pem_private_key(pk_string, password=None, backend=default_backend())
 
     pkb = p_key.private_bytes(