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)