Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion dbldatagen/data_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -1913,7 +1913,7 @@ def scriptMerge(
result = "\n".join(results)

if asHtml:
result = HtmlUtils.formatCodeAsHtml(results)
result = HtmlUtils.formatCodeAsHtml(result)

return result

Expand Down
5 changes: 3 additions & 2 deletions dbldatagen/datasets/dataset_provider.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
This file defines the DatasetProvider class
"""


class DatasetProvider(ABC):
"""
The DatasetProvider class acts as a base class for all dataset providers
Expand Down Expand Up @@ -206,7 +207,7 @@ def getTableGenerator(self, sparkSession: SparkSession, *, tableName: str|None=N
raise NotImplementedError("Base data provider does not provide any table generation specifications!")

@abstractmethod
def getAssociatedDataset(self, sparkSession: SparkSession, *, tableName: str|None=None, rows: int=-1, partitions: int=-1,
def getAssociatedDataset(self, sparkSession: SparkSession, *, tableName: str | None=None, rows: int=-1, partitions: int=-1,
**options: dict[str, Any]) -> DataGenerator:
"""
Gets associated datasets that are used in conjunction with the provider datasets.
Expand Down Expand Up @@ -240,7 +241,7 @@ def allowed_options(options: list[str]|None =None) -> Callable[[Callable], Calla

def decorator(func: Callable) -> Callable:
@functools.wraps(func)
def wrapper(*args, **kwargs) -> Callable: # noqa: ANN002
def wrapper(*args, **kwargs) -> Callable:
bad_options = [keyword_arg for keyword_arg in kwargs
if keyword_arg not in DEFAULT_OPTIONS and keyword_arg not in options]

Expand Down
Loading