# Databricks connector

Union.ai can be integrated with the [Databricks](https://www.databricks.com/) service,
enabling you to submit Spark jobs to the Databricks platform.

## Installation

The Databricks connector comes bundled with the Spark plugin. To install the Spark plugin, run the following command:

```shell
$ pip install flytekitplugins-spark
```

## Example usage

For an example query, see [Databricks connector example usage](https://www.union.ai/docs/v1/union/integrations/connectors/databricks-connector/databricks-connector-example-usage-union/page.md)

## Local testing

To test the Databricks connector locally, create a class for the connector task that inherits from
[AsyncConnectorExecutorMixin](https://github.com/flyteorg/flytekit/blob/1bc8302bb7a6cf4c7048a7f93627ee25fc6b88c4/flytekit/extend/backend/base_connector.py#L354).
This mixin can handle asynchronous tasks and allows the SDK to mimic the system's behavior in calling the connector.

For more information, see [Testing connectors locally](https://www.union.ai/docs/v1/union/integrations/connectors/page.md#testing-your-connector-locally).

> [!NOTE]
> In some cases, you will need to store credentials in your local environment when testing locally.

## Subpages

- [Running Spark on Databricks](https://www.union.ai/docs/v1/union/integrations/connectors/databricks-connector/databricks-connector-example-usage-union/page.md)

---
**Source**: https://github.com/unionai/unionai-docs/blob/main/content/integrations/connectors/databricks-connector/_index.md
**HTML**: https://www.union.ai/docs/v1/union/integrations/connectors/databricks-connector/