test var handling

tests/tlm/constants.py

@@ -1,10 +1,12 @@
+from typing import List
+
 from cleanlab_studio.internal.constants import _VALID_TLM_MODELS
 
 # Test TLM
-TEST_PROMPT = "What is the capital of France?"
-TEST_RESPONSE = "Paris"
-TEST_PROMPT_BATCH = ["What is the capital of France?", "What is the capital of Ukraine?"]
-TEST_RESPONSE_BATCH = ["Paris", "Kyiv"]
+TEST_PROMPT: str = "What is the capital of France?"
+TEST_RESPONSE: str = "Paris"
+TEST_PROMPT_BATCH: List[str] = ["What is the capital of France?", "What is the capital of Ukraine?"]
+TEST_RESPONSE_BATCH: List[str] = ["Paris", "Kyiv"]
 
 # Test validation tests for TLM
 MAX_PROMPT_LENGTH_TOKENS: int = 70_000
@@ -14,6 +16,12 @@
 CHARACTERS_PER_TOKEN: int = 4
 
 # Property tests for TLM
-excluded_tlm_models = ["claude-3-sonnet", "claude-3.5-sonnet"]
-VALID_TLM_MODELS = [model for model in _VALID_TLM_MODELS if model not in excluded_tlm_models]
-MODELS_WITH_NO_PERPLEXITY_SCORE = ["claude-3-haiku", "claude-3-sonnet", "claude-3.5-sonnet"]
+excluded_tlm_models: List[str] = ["claude-3-sonnet", "claude-3.5-sonnet"]
+VALID_TLM_MODELS: List[str] = [
+ model for model in _VALID_TLM_MODELS if model not in excluded_tlm_models
+]
+MODELS_WITH_NO_PERPLEXITY_SCORE: List[str] = [
+ "claude-3-haiku",
+ "claude-3-sonnet",
+ "claude-3.5-sonnet",
+]

tests/tlm/test_properties.py

@@ -19,7 +19,7 @@
 )
 from tests.tlm.test_prompt import is_tlm_response
 
-test_prompt = make_text_unique(str(TEST_PROMPT))
+test_prompt = [make_text_unique(prompt) for prompt in [TEST_PROMPT]][0]
 test_prompt_batch = [make_text_unique(prompt) for prompt in TEST_PROMPT_BATCH]