minor changes

Author: avinash2692

test/backends/test_watsonx.py

@@ -15,36 +15,42 @@
 @pytest.fixture(scope="module")
 def backend():
     """Shared Watson backend for all tests in this module."""
-    return WatsonxAIBackend(
+    if os.environ.get("CICD") == 1:
+        pytest.skip("Skipping watsonx tests.")
+    else:
+        return WatsonxAIBackend(
         model_id="ibm/granite-3-3-8b-instruct",
         formatter=TemplateFormatter(model_id="ibm-granite/granite-3.3-8b-instruct"),
     )
 
 
 @pytest.fixture(scope="function")
-def session(backend):
-    """Fresh Watson session for each test."""
-    session = MelleaSession(backend, ctx=LinearContext(is_chat_context=True))
-    yield session
-    session.reset()
+def session(backend: WatsonxAIBackend):
+    if os.environ.get("CICD") == 1:
+        pytest.skip("Skipping watsonx tests.")
+    else:
+        """Fresh Watson session for each test."""
+        session = MelleaSession(backend, ctx=LinearContext(is_chat_context=True))
+        yield session
+        session.reset()
 
 
 @pytest.mark.qualitative
-def test_instruct(session):
+def test_instruct(session: MelleaSession):
     result = session.instruct("Compute 1+1.")
     assert isinstance(result, ModelOutputThunk)
     assert "2" in result.value  # type: ignore
 
 
 @pytest.mark.qualitative
-def test_multiturn(session):
+def test_multiturn(session: MelleaSession):
     session.instruct("What is the capital of France?")
     answer = session.instruct("Tell me the answer to the previous question.")
     assert "Paris" in answer.value  # type: ignore
 
 
 @pytest.mark.qualitative
-def test_format(session):
+def test_format(session: MelleaSession):
     class Person(pydantic.BaseModel):
         name: str
         # it does not support regex patterns in json schema
@@ -78,7 +84,7 @@ class Email(pydantic.BaseModel):
 
 
 @pytest.mark.qualitative
-def test_generate_from_raw(session):
+def test_generate_from_raw(session: MelleaSession):
     prompts = ["what is 1+1?", "what is 2+2?", "what is 3+3?", "what is 4+4?"]
 
     results = session.backend._generate_from_raw(

test/stdlib_basics/test_session.py

@@ -1,3 +1,5 @@
+import os
+
 import pytest
 
 from mellea.stdlib.base import ModelOutputThunk