Fix unit tests

Author: stefannica

src/zenml/deployers/server/models.py

@@ -60,9 +60,10 @@ class BaseDeploymentInvocationRequest(BaseModel):
     timeout: int = Field(
         default=300, title="The timeout for the pipeline execution."
     )
-    use_in_memory: bool = Field(
+    skip_artifact_materialization: bool = Field(
         default=False,
-        title="Whether to keep outputs in memory for fast access.",
+        title="Whether to keep outputs in memory for fast access instead of "
+        "storing them as artifacts.",
     )
 
 

src/zenml/deployers/server/service.py

@@ -208,7 +208,7 @@ def execute_pipeline(
                 placeholder_run=placeholder_run,
                 deployment_snapshot=deployment_snapshot,
                 resolved_params=parameters,
-                use_in_memory=request.use_in_memory,
+                skip_artifact_materialization=request.skip_artifact_materialization,
             )
 
             # Map outputs using fast (in-memory) or slow (artifact) path
@@ -370,7 +370,7 @@ def _execute_with_orchestrator(
         placeholder_run: PipelineRunResponse,
         deployment_snapshot: PipelineSnapshotResponse,
         resolved_params: Dict[str, Any],
-        use_in_memory: bool,
+        skip_artifact_materialization: bool,
     ) -> Optional[Dict[str, Dict[str, Any]]]:
         """Run the snapshot via the orchestrator and return the concrete run.
 
@@ -379,7 +379,8 @@ def _execute_with_orchestrator(
             deployment_snapshot: The deployment snapshot to execute the pipeline
                 on.
             resolved_params: Normalized pipeline parameters.
-            use_in_memory: Whether runtime should capture in-memory outputs.
+            skip_artifact_materialization: Whether runtime should skip artifact
+                materialization.
 
         Returns:
             The in-memory outputs of the execution.
@@ -400,7 +401,7 @@ def _execute_with_orchestrator(
             request_id=str(uuid4()),
             snapshot=deployment_snapshot,
             parameters=resolved_params,
-            skip_artifact_materialization=use_in_memory,
+            skip_artifact_materialization=skip_artifact_materialization,
         )
 
         captured_outputs: Optional[Dict[str, Dict[str, Any]]] = None

src/zenml/hooks/hook_validators.py

@@ -22,7 +22,7 @@
     Union,
 )
 
-from pydantic import ConfigDict, PydanticSchemaGenerationError, ValidationError
+from pydantic import ConfigDict, ValidationError
 
 from zenml.config.source import Source
 from zenml.exceptions import HookValidationException
@@ -69,7 +69,7 @@ def resolve_and_validate_hook(
 
     # Validate hook arguments
     try:
-        hook_args = ()
+        hook_args: Tuple[Any, ...] = ()
         if allow_exception_arg:
             hook_args = (Exception(),)
         hook_kwargs = hook_kwargs or {}

src/zenml/steps/step_context.py

@@ -62,13 +62,11 @@ def get_step_context() -> "StepContext":
 
 
 def get_or_create_run_context() -> "RunContext":
-    """Get the context of the currently running pipeline.
+    """Get or create the context of the currently running pipeline.
 
     Returns:
         The context of the currently running pipeline.
     """
-    if RunContext._exists():
-        return RunContext()
     return RunContext()
 
 

tests/integration/functional/deployers/serving/conftest.py renamed to tests/integration/functional/deployers/server/conftest.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Test-specific fixtures for serving integration tests."""
+"""Test-specific fixtures for deployment integration tests."""
 
 from types import SimpleNamespace
 from typing import Generator, Tuple

tests/integration/functional/deployers/serving/test_app_endpoints.py renamed to tests/integration/functional/deployers/server/test_app_endpoints.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Integration tests for FastAPI serving application endpoints."""
+"""Integration tests for FastAPI deployment application endpoints."""
 
 import importlib
 from types import ModuleType, SimpleNamespace
@@ -22,7 +22,7 @@
 from fastapi.testclient import TestClient
 from pydantic import BaseModel
 
-import zenml.deployers.server.app as serving_app
+import zenml.deployers.server.app as deployment_app
 from zenml.deployers.server.models import (
     BaseDeploymentInvocationRequest,
     BaseDeploymentInvocationResponse,
@@ -190,7 +190,7 @@ def client_service_pair(
     Yields:
         A tuple containing the FastAPI client, the stub service, and the reloaded app.
     """
-    reloaded_app = importlib.reload(serving_app)
+    reloaded_app = importlib.reload(deployment_app)
     service = StubDeploymentService(str(uuid4()))
 
     monkeypatch.setenv("ZENML_DEPLOYMENT_ID", str(service.deployment.id))
@@ -346,7 +346,7 @@ def test_cleanup_called_on_shutdown(
         ],
     ) -> None:
         """Trigger service cleanup when the application shuts down."""
-        reloaded_app = importlib.reload(serving_app)
+        reloaded_app = importlib.reload(deployment_app)
         service = StubDeploymentService(str(uuid4()))
         monkeypatch.setenv("ZENML_DEPLOYMENT_ID", str(service.deployment.id))
         monkeypatch.setattr(

tests/unit/deployers/server/__init__.py

@@ -11,4 +11,4 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Unit tests for serving functionality."""
+"""Unit tests for deployment functionality."""

tests/unit/deployers/server/conftest.py

@@ -1,4 +1,4 @@
-"""Test-specific fixtures for serving unit tests."""
+"""Test-specific fixtures for deployment unit tests."""
 
 from types import SimpleNamespace
 from typing import Iterator, Tuple

tests/unit/deployers/server/test_app.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Unit tests for serving app functionality."""
+"""Unit tests for deployment app functionality."""
 
 from __future__ import annotations
 
@@ -57,7 +57,7 @@ class MockWeatherRequest(BaseModel):
 
 @pytest.fixture
 def mock_service(mocker: MockerFixture) -> PipelineDeploymentService:
-    """Mock pipeline serving service configured for the app tests."""
+    """Mock pipeline deployment service configured for the app tests."""
 
     service = cast(
         PipelineDeploymentService,
@@ -114,7 +114,7 @@ def mock_service(mocker: MockerFixture) -> PipelineDeploymentService:
     return service
 
 
-class TestServingAppRoutes:
+class TestDeploymentAppRoutes:
     """Test FastAPI app routes."""
 
     def test_root_endpoint(
@@ -236,7 +236,7 @@ def test_get_pipeline_service_returns_current_instance(
         assert get_pipeline_service() is mock_service
 
 
-class TestServingAppInvoke:
+class TestDeploymentAppInvoke:
     """Test pipeline invocation via FastAPI."""
 
     def test_invoke_endpoint_executes_service(
@@ -255,7 +255,7 @@ def test_invoke_endpoint_executes_service(
         mock_service.execute_pipeline.assert_called_once()
         request_arg = mock_service.execute_pipeline.call_args.args[0]
         assert request_arg.parameters.city == "Paris"
-        assert request_arg.use_in_memory is False
+        assert request_arg.skip_artifact_materialization is False
 
     def test_invoke_endpoint_validation_error(
         self, mock_service: PipelineDeploymentService
@@ -299,7 +299,7 @@ def test_verify_token_with_auth_disabled(
         assert verify_token(None) is None
 
 
-class TestServingAppLifecycle:
+class TestDeploymentAppLifecycle:
     """Test app lifecycle management."""
 
     def test_lifespan_test_mode(self, monkeypatch: pytest.MonkeyPatch) -> None:
@@ -358,7 +358,7 @@ async def _run() -> None:
         asyncio.run(_run())
 
 
-class TestServingAppErrorHandling:
+class TestDeploymentAppErrorHandling:
     """Test app error handling."""
 
     def test_value_error_handler(self) -> None:

tests/unit/deployers/server/test_parameter_flow.py

@@ -11,7 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""Comprehensive test for parameter resolution and flow in serving."""
+"""Comprehensive test for parameter resolution and flow in deployment."""
 
 from unittest.mock import MagicMock
 
@@ -24,7 +24,7 @@ class TestOutputRecording:
     """Test output recording and retrieval functionality."""
 
     @pytest.fixture(autouse=True)
-    def setup_serving_state(self):
+    def setup_deployment_state(self):
         """Set up deployment state for each test."""
         runtime.stop()
         yield