fix type hints

Author: strickvl

llm-complete-guide/pipelines/distilabel_generation.py

@@ -48,7 +48,7 @@
 @pipeline(model=model_definition)
 def generate_synthetic_data():
     train_dataset, test_dataset = load_hf_dataset()
-    train_pii_results, test_pii_results = eval_pii(
+    _, _, _ = eval_pii(
         train_dataset=train_dataset,
         test_dataset=test_dataset,
     )

llm-complete-guide/steps/distilabel_generate_queries.py

@@ -45,7 +45,7 @@ def generate_synthetic_queries(
 
     with Pipeline(name="generate_embedding_queries") as pipeline:
         load_dataset = LoadDataFromHub(
-            # num_examples=20,  # use this for demo purposes
+            num_examples=40,  # use this for demo purposes
             output_mappings={"page_content": "anchor"},
         )
         generate_sentence_pair = GenerateSentencePair(

llm-complete-guide/steps/eval_pii.py

@@ -272,7 +272,7 @@ def eval_pii(
 ) -> Tuple[
     Annotated[Dict[str, Dict], "train_results"],
     Annotated[Dict[str, Dict], "test_results"],
-    Annotated[Image, "PII chart"],
+    Annotated[Image.Image, "PII chart"],
 ]:
     detector = PIIDetector()
     train_results = detector.scan_dataset(