diff --git a/dataherald/scripts/populate_dialect_db_connection.py b/dataherald/scripts/populate_dialect_db_connection.py index 7f15cb6c..290461cf 100644 --- a/dataherald/scripts/populate_dialect_db_connection.py +++ b/dataherald/scripts/populate_dialect_db_connection.py @@ -13,7 +13,9 @@ database_connections = storage.find_all("database_connections") for database_connection in database_connections: if not database_connection.get("dialect"): - decrypted_uri = fernet_encrypt.decrypt(database_connection["connection_uri"]) + decrypted_uri = fernet_encrypt.decrypt( + database_connection["connection_uri"] + ) dialect_prefix = DatabaseConnection.validate_uri(decrypted_uri) dialect = DatabaseConnection.set_dialect(dialect_prefix) storage.update_or_create(