"""``JSONDataset`` saves data to a JSON file using an underlying
filesystem (e.g.: local, S3, GCS). It uses native json to handle the JSON file.
The ``JSONDataset`` is part of Kedro Experiment Tracking. The dataset is versioned by default.
"""
import json
from typing import NoReturn
from kedro.io.core import DatasetError, get_filepath_str
from kedro_datasets._typing import JSONTrackingPreview
from kedro_datasets.json import json_dataset
[docs]
class JSONDataset(json_dataset.JSONDataset):
"""``JSONDataset`` saves data to a JSON file using an underlying
filesystem (e.g.: local, S3, GCS). It uses native json to handle the JSON file.
The ``JSONDataset`` is part of Kedro Experiment Tracking.
The dataset is write-only and it is versioned by default.
Example usage for the
`YAML API <https://docs.kedro.org/en/stable/data/\
data_catalog_yaml_examples.html>`_:
.. code-block:: yaml
cars:
type: tracking.JSONDataset
filepath: data/09_tracking/cars.json
Example usage for the
`Python API <https://docs.kedro.org/en/stable/data/\
advanced_data_catalog_usage.html>`_:
.. code-block:: pycon
>>> from kedro_datasets.tracking import JSONDataset
>>>
>>> data = {"col1": 1, "col2": 0.23, "col3": 0.002}
>>>
>>> dataset = JSONDataset(filepath=tmp_path / "test.json")
>>> dataset.save(data)
"""
versioned = True
[docs]
def load(self) -> NoReturn:
raise DatasetError(f"Loading not supported for '{self.__class__.__name__}'")
[docs]
def preview(self) -> JSONTrackingPreview: # type: ignore[override]
"Load the JSON tracking dataset used in Kedro-viz experiment tracking."
load_path = get_filepath_str(self._get_load_path(), self._protocol)
with self._fs.open(load_path, **self._fs_open_args_load) as fs_file:
return JSONTrackingPreview(json.load(fs_file))
