Skip to content

Commit

Permalink
fix save method error to kedro.io.DatasetError in spark.GBQQueryDataset
Browse files Browse the repository at this point in the history
  • Loading branch information
abhi8893 committed Jan 1, 2025
1 parent d70d231 commit 6d7b39a
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions kedro-datasets/kedro_datasets/spark/spark_gbq_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from copy import deepcopy
from typing import Any, NoReturn

from kedro.io import AbstractDataset
from kedro.io import AbstractDataset, DatasetError
from py4j.protocol import Py4JJavaError
from pyspark.sql import DataFrame

Expand Down Expand Up @@ -166,9 +166,7 @@ def load(self) -> DataFrame:
return read_obj.load(**self._get_spark_load_args())

def save(self, data: None) -> NoReturn:
raise NotImplementedError(
"Save method is not implemented for spark.GBQQueryDataset"
)
raise DatasetError("'save' is not supported on GBQQueryDataset")

def _describe(self) -> dict[str, Any]:
return {
Expand Down

0 comments on commit 6d7b39a

Please sign in to comment.