From 1484fb59b156632f8c625971d41494317700316d Mon Sep 17 00:00:00 2001 From: kdziedzic68 Date: Tue, 12 Nov 2024 14:41:40 +0100 Subject: [PATCH] ruff formating --- examples/evaluation/dataset-generator/generate.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/examples/evaluation/dataset-generator/generate.py b/examples/evaluation/dataset-generator/generate.py index cdb983df..f0733e9b 100644 --- a/examples/evaluation/dataset-generator/generate.py +++ b/examples/evaluation/dataset-generator/generate.py @@ -6,7 +6,14 @@ @hydra.main(config_path="config", config_name="generate", version_base="3.2") -def main(config: DictConfig): +def main(config: DictConfig) -> None: + """ + A main function for dataset generation example + Args: + config - configuration + Returns: + None + """ TOPICS = ["conspiracy theories", "machine learning"] generation_pipeline = DatasetGenerationPipeline(config=config)