> ## Documentation Index
> Fetch the complete documentation index at: https://docs.datazone.co/llms.txt
> Use this file to discover all available pages before exploring further.

# Toolkit

> The Toolkit is a collection of tools and utilities that help you manage your notebooks in Datazone.

## Overview

The Toolkit provides utilities to help you work with data in Datazone notebooks. Currently, it includes the Dataset class for accessing datasets.

## Dataset

The Dataset class allows you to easily load datasets into your notebooks as Pandas or PySpark DataFrames.

```python theme={null}
from datazone import Dataset

dataset = Dataset(id="<dataset-id>")
# or
dataset = Dataset(alias="<dataset-alias>")
```

You can also load a dataset by providing a specific branch name:

```python theme={null}
from datazone import Dataset

dataset = Dataset(id="<dataset-id>", branch="<branch-name>")
```

Thanks to the Dataset class, you can now easily load datasets into your notebooks as Pandas or PySpark DataFrames.

```python theme={null}
from datazone import Dataset

pandas_df = dataset.get_pandas_df()
# or
pyspark_df = dataset.get_pyspark_df()
```

<Warning>`get_pyspark_df()` is only available in the PySpark kernel.</Warning>

<Frame>
  <img src="https://mintcdn.com/datazone/YFojkVY1avf3O4H9/images/light/notebook/dataset-cell-example.png?fit=max&auto=format&n=YFojkVY1avf3O4H9&q=85&s=8d49eda94b9c49c4df445c100fefabf5" alt="dataset-cell-example" width="2302" height="828" data-path="images/light/notebook/dataset-cell-example.png" />
</Frame>

## Variable

You can access the Variables from the kernel environment using the `Variable` class.

```python theme={null}
from datazone import Variable

variable = Variable(key="<variable-name>")
```

### Example

```python theme={null}
from datazone import Variable

api_key = Variable(key="OPENAI_API_KEY")
client = OpenAI(api_key=str(api_key))
...
```

Also you can check the [Variables](/reference/development/variables)