Added import tests for num2words and PyYAML

giskard/llm/client/dummy_client.py

@@ -0,0 +1,26 @@
+from .base import LLMClient, ChatMessage
+from typing import Sequence, Optional
+
+class DummyLLMClient(LLMClient):
+    def complete(
+        self,
+        messages: Sequence[ChatMessage],
+        temperature: float = 1,
+        max_tokens: Optional[int] = None,
+        caller_id: Optional[str] = None,
+        seed: Optional[int] = None,
+        format=None,
+    ) -> ChatMessage:
+        # Simple echo implementation
+        last_message = messages[-1] if messages else ChatMessage(role="user", content="Hello")
+        return ChatMessage(role="assistant", content=f"Echo: {last_message.content}")
+
+    def get_config(self) -> dict:
+        return {"name": "DummyLLMClient", "version": "1.0"}
+
+if __name__ == "__main__":
+    client = DummyLLMClient()
+    messages = [ChatMessage(role="user", content="Hi there!")]
+    response = client.complete(messages)
+    print("Response:", response)
+    print("Config:", client.get_config())

giskard/testing/tests/drift.py

@@ -9,8 +9,7 @@
 
 import numpy as np
 import pandas as pd
-from scipy.stats import chi2, ks_2samp
-from scipy.stats.stats import Ks_2sampResult, wasserstein_distance
+from scipy.stats import chi2, ks_2samp, wasserstein_distance
 
 from giskard.core.test_result import TestMessage, TestMessageLevel, TestResult
 from giskard.datasets.base import Dataset
@@ -99,7 +98,7 @@ def _calculate_drift_psi(actual_series, reference_series, max_categories):
     return total_psi, pd.DataFrame(output_data)
 
 
-def _calculate_ks(actual_series, reference_series) -> Ks_2sampResult:
+def _calculate_ks(actual_series, reference_series) -> typing.Any:
     return ks_2samp(reference_series, actual_series)
 
 

test_num2words_import.py

@@ -0,0 +1,5 @@
+try:
+    from num2words import num2words
+    print("num2words is installed and importable.")
+except ImportError as e:
+    print("ImportError:", e)

test_yaml_import.py

@@ -0,0 +1,5 @@
+try:
+    import yaml
+    print("PyYAML is installed and importable.")
+except ImportError as e:
+    print("ImportError:", e)