ruff formating

deepsense-ai · Nov 12, 2024 · 1484fb5 · 1484fb5
1 parent 56f3611
commit 1484fb5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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)