wafer-cli 0.2.13__tar.gz → 0.2.14__tar.gz

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.13
+Version: 0.2.14
 Summary: CLI tool for running commands on remote GPUs and GPU kernel optimization agent
 Requires-Python: >=3.11
 Requires-Dist: typer>=0.12.0

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wafer-cli"
-version = "0.2.13"
+version = "0.2.14"
 description = "CLI tool for running commands on remote GPUs and GPU kernel optimization agent"
 requires-python = ">=3.11"
 dependencies = [
@@ -76,9 +76,19 @@ ignore = [
 ]
 
 [tool.ruff.lint.per-file-ignores]
-"tests/**/*.py" = ["ANN201"]  # Don't require return type annotations in tests
-"wafer/evaluate.py" = ["PLR0915", "PLR1702", "E402"]  # complex deployment flows - TODO: refactor
+"tests/**/*.py" = ["ANN001", "ANN201", "ANN202", "ANN204"]  # Don't require type annotations in tests
+"wafer/evaluate.py" = ["PLR0915", "PLR1702", "E402", "PLW2901", "ASYNC221"]  # complex deployment flows - TODO: refactor
 "wafer/output.py" = ["ANN401"]  # Output collector uses **kwargs for flexible event data
+"wafer/autotuner.py" = ["PLR0915", "PLR1702", "B007", "B904"]  # complex sweep logic - TODO: refactor
+"wafer/ncu_analyze.py" = ["PLR0915", "PLR1702"]  # complex parsing logic - TODO: refactor
+"wafer/cli.py" = ["PLR0915"]  # CLI commands can be long - TODO: refactor
+"wafer/targets.py" = ["PLR1702"]  # complex target init flow - TODO: refactor
+"wafer/workspaces.py" = ["PLR1702"]  # SSE streaming has nested blocks - TODO: refactor
+"wafer/kernel_scope.py" = ["ANN001", "ANN202", "PLR0913"]  # complex filtering logic
+"wafer/api_client.py" = ["PLR0913"]  # API client needs many params
+"wafer/rocprof_compute.py" = ["PLR0913", "B904"]  # profiler commands need many params
+"wafer/rocprof_sdk.py" = ["PLR0913"]  # profiler commands need many params
+"wafer/rocprof_systems.py" = ["PLR0913"]  # profiler commands need many params
 
 [tool.ruff.lint.pylint]
 max-args = 7             # Max function arguments (Tiger Style: few parameters)

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_analytics.py

@@ -11,12 +11,9 @@ Tests cover:
 Run with: PYTHONPATH=apps/wafer-cli uv run pytest apps/wafer-cli/tests/test_analytics.py -v
 """
 
-import json
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
-import pytest
-
 
 class TestAnalyticsInit:
     """Test analytics initialization."""
@@ -403,16 +400,17 @@ class TestCliCallback:
     def test_cli_callback_tracks_command(self) -> None:
         """CLI callback should track command execution."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
 
         runner = CliRunner()
 
         with patch("wafer.analytics.init_analytics") as mock_init, \
-             patch("wafer.analytics.track_command") as mock_track:
+             patch("wafer.analytics.track_command"):
             mock_init.return_value = True
 
             # Run a simple command that doesn't require auth
-            result = runner.invoke(app, ["guide"])
+            runner.invoke(app, ["guide"])
 
             # Analytics should be initialized
             mock_init.assert_called()
@@ -423,6 +421,7 @@ class TestCliCallback:
     def test_cli_help_does_not_crash_analytics(self) -> None:
         """CLI --help should not crash due to analytics."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
 
         runner = CliRunner()
@@ -435,6 +434,7 @@ class TestCliCallback:
     def test_cli_subcommand_help_works(self) -> None:
         """Subcommand --help should work with analytics."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
 
         runner = CliRunner()
@@ -452,6 +452,7 @@ class TestLoginLogoutAnalytics:
         from unittest.mock import MagicMock
 
         from typer.testing import CliRunner
+
         from wafer.cli import app
 
         runner = CliRunner()
@@ -466,7 +467,7 @@ class TestLoginLogoutAnalytics:
              patch("wafer.analytics.track_login") as mock_track_login, \
              patch("wafer.analytics.init_analytics", return_value=True):
 
-            result = runner.invoke(app, ["login", "--token", "test-token"])
+            runner.invoke(app, ["login", "--token", "test-token"])
 
             # track_login should be called
             mock_track_login.assert_called_once_with("test-user-id", "test@example.com")
@@ -474,6 +475,7 @@ class TestLoginLogoutAnalytics:
     def test_logout_calls_track_logout(self) -> None:
         """Logout command should call track_logout."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
 
         runner = CliRunner()

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_kernel_scope_cli.py

@@ -8,7 +8,6 @@ Run with: PYTHONPATH=apps/wafer-cli uv run pytest apps/wafer-cli/tests/test_kern
 import json
 import re
 from pathlib import Path
-from unittest.mock import patch, MagicMock
 
 import pytest
 from typer.testing import CliRunner
@@ -20,7 +19,6 @@ from wafer.kernel_scope import (
     targets_command,
 )
 
-
 runner = CliRunner()
 
 

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_nsys_analyze.py

@@ -7,16 +7,12 @@ Tests the nsys_analyze module including:
 - API fallback logic
 """
 
-import json
-import os
-import platform
 from pathlib import Path
-from unittest.mock import MagicMock, patch
+from unittest.mock import patch
 
 import pytest
 
 from wafer.nsys_analyze import (
-    NSYSCheckResult,
     _find_nsys,
     _get_install_command,
     _get_platform,
@@ -247,7 +243,7 @@ class TestAnalyzeNsysProfile:
         with patch("wafer.nsys_analyze._find_nsys", return_value=None):
             with patch("wafer.nsys_analyze._analyze_remote_api") as mock_remote:
                 mock_remote.return_value = '{"success": true}'
-                result = analyze_nsys_profile(test_file)
+                analyze_nsys_profile(test_file)
                 mock_remote.assert_called_once()
 
     def test_uses_local_when_nsys_installed(self, tmp_path: Path) -> None:
@@ -258,7 +254,7 @@ class TestAnalyzeNsysProfile:
         with patch("wafer.nsys_analyze._find_nsys", return_value="/usr/bin/nsys"):
             with patch("wafer.nsys_analyze._analyze_local") as mock_local:
                 mock_local.return_value = '{"success": true}'
-                result = analyze_nsys_profile(test_file)
+                analyze_nsys_profile(test_file)
                 mock_local.assert_called_once()
 
     def test_respects_remote_flag(self, tmp_path: Path) -> None:
@@ -269,7 +265,7 @@ class TestAnalyzeNsysProfile:
         with patch("wafer.nsys_analyze._find_nsys", return_value="/usr/bin/nsys"):
             with patch("wafer.nsys_analyze._analyze_remote_api") as mock_remote:
                 mock_remote.return_value = '{"success": true}'
-                result = analyze_nsys_profile(test_file, remote=True)
+                analyze_nsys_profile(test_file, remote=True)
                 mock_remote.assert_called_once()
 
     def test_uses_workspace_when_specified(self, tmp_path: Path) -> None:
@@ -279,7 +275,7 @@ class TestAnalyzeNsysProfile:
 
         with patch("wafer.nsys_analyze._analyze_workspace") as mock_workspace:
             mock_workspace.return_value = '{"success": true}'
-            result = analyze_nsys_profile(test_file, target="workspace:abc123")
+            analyze_nsys_profile(test_file, target="workspace:abc123")
             mock_workspace.assert_called_once_with(test_file, "abc123", False)
 
     def test_uses_target_when_specified(self, tmp_path: Path) -> None:
@@ -289,5 +285,5 @@ class TestAnalyzeNsysProfile:
 
         with patch("wafer.nsys_analyze._analyze_remote_direct") as mock_direct:
             mock_direct.return_value = '{"success": true}'
-            result = analyze_nsys_profile(test_file, target="vultr-b200")
+            analyze_nsys_profile(test_file, target="vultr-b200")
             mock_direct.assert_called_once_with(test_file, "vultr-b200", False)

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_nsys_profile.py

@@ -1,14 +1,15 @@
 """Tests for NSYS profile command."""
 
+from unittest.mock import patch
+
 import pytest
-from unittest.mock import Mock, patch, MagicMock
+
 from wafer.nsys_profile import (
     NSYSProfileOptions,
     NSYSProfileResult,
-    profile_local,
-    profile_workspace,
-    profile_and_analyze,
     _build_nsys_command,
+    profile_and_analyze,
+    profile_local,
 )
 
 
@@ -136,6 +137,7 @@ class TestCLIIntegration:
     def test_profile_command_help(self):
         """Verify profile command exists and has help text."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
         
         runner = CliRunner()
@@ -148,6 +150,7 @@ class TestCLIIntegration:
     def test_profile_command_requires_command(self):
         """Verify profile command requires a command argument."""
         from typer.testing import CliRunner
+
         from wafer.cli import app
         
         runner = CliRunner()

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_rocprof_compute_integration.py

@@ -10,9 +10,8 @@ Follows similar testing patterns from the codebase.
 """
 
 import json
-import subprocess
 from pathlib import Path
-from unittest.mock import patch, Mock
+from unittest.mock import Mock, patch
 
 import pytest
 
@@ -139,9 +138,10 @@ class TestIntegrationWithCore:
         tmp_path: Path
     ) -> None:
         """Test CLI uses wafer-core launch_gui."""
-        from wafer.rocprof_compute import launch_gui
         from wafer_core.lib.rocprofiler.compute.types import CheckResult
 
+        from wafer.rocprof_compute import launch_gui
+
         test_folder = tmp_path / "results"
         test_folder.mkdir()
 
@@ -173,9 +173,10 @@ class TestCommandFormat:
         tmp_path: Path
     ) -> None:
         """Test generated command uses correct CLI flags when using external binary."""
-        from wafer.rocprof_compute import launch_gui
         from wafer_core.lib.rocprofiler.compute.types import CheckResult, LaunchResult
 
+        from wafer.rocprof_compute import launch_gui
+
         test_folder = tmp_path / "results"
         test_folder.mkdir()
 

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/tests/test_wevin_cli.py

@@ -6,10 +6,8 @@ because they require complex integration tests to execute.
 """
 
 import json
-import os
 import tempfile
 from pathlib import Path
-from unittest.mock import patch
 
 from wafer_core.rollouts import Endpoint, FileSessionStore
 from wafer_core.rollouts.dtypes import Message
@@ -125,7 +123,7 @@ def test_list_sessions_returns_metadata_only():
         endpoint = Endpoint(provider="anthropic", model="claude-sonnet-4.5")
         env_config = EnvironmentConfig(type="localfs")
         
-        async def run_test():
+        async def run_test() -> None:
             # Create a session with messages
             session = await session_store.create(endpoint=endpoint, environment=env_config)
             
@@ -182,8 +180,9 @@ def test_get_session_json_output():
     
     Tests the CLI command used by wevin-extension to load session messages.
     """
-    import trio
     from dataclasses import asdict
+
+    import trio
     from wafer_core.rollouts import EnvironmentConfig
     
     with tempfile.TemporaryDirectory() as tmpdir:
@@ -193,7 +192,7 @@ def test_get_session_json_output():
         endpoint = Endpoint(provider="anthropic", model="claude-sonnet-4.5")
         env_config = EnvironmentConfig(type="localfs")
         
-        async def run_test():
+        async def run_test() -> None:
             session = await session_store.create(endpoint=endpoint, environment=env_config)
             
             # Add some messages
@@ -240,7 +239,7 @@ def test_get_session_not_found():
     with tempfile.TemporaryDirectory() as tmpdir:
         session_store = FileSessionStore(base_dir=Path(tmpdir))
         
-        async def run_test():
+        async def run_test() -> None:
             session, err = await session_store.get("nonexistent-session-id")
             assert session is None
             assert err is not None
@@ -289,7 +288,7 @@ def test_session_resume_loads_context():
         endpoint = Endpoint(provider="anthropic", model="claude-sonnet-4.5")
         env_config = EnvironmentConfig(type="localfs")
         
-        async def run_test():
+        async def run_test() -> None:
             # Create initial session with messages
             session = await session_store.create(endpoint=endpoint, environment=env_config)
             
@@ -335,8 +334,9 @@ def test_cli_parameters_accepted():
     - max_tool_fails: Exit after N failures
     - max_turns: Limit conversation turns
     """
-    from wafer.wevin_cli import main as wevin_main
     import inspect
+
+    from wafer.wevin_cli import main as wevin_main
     
     sig = inspect.signature(wevin_main)
     params = sig.parameters
@@ -370,15 +370,16 @@ def test_streaming_frontend_session_start_resumed_session():
     
     Edge case: When session_id is known upfront (from --resume), emit immediately.
     """
+    from io import StringIO
+
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
-    from io import StringIO
-    import sys
     
     # Capture stdout
     captured_output = StringIO()
     
-    async def run_test():
+    async def run_test() -> None:
         frontend = StreamingChunkFrontend(
             session_id="test-session-123",
             model="claude-sonnet-4.5"
@@ -386,8 +387,8 @@ def test_streaming_frontend_session_start_resumed_session():
         
         # Mock _emit to capture output
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
             # Also print to verify JSON format
             print(json.dumps(obj), file=captured_output)
@@ -418,14 +419,15 @@ def test_streaming_frontend_session_start_no_session_id():
     Edge case: New session - session_id not known until after run_interactive.
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         frontend = StreamingChunkFrontend(session_id=None, model=None)
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -444,15 +446,16 @@ def test_streaming_frontend_emit_session_start_new_session():
     Edge case: Session created during run_interactive, emit after first state.
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         # Frontend starts without session_id (new session)
         frontend = StreamingChunkFrontend(session_id=None, model="claude-sonnet-4.5")
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -480,15 +483,16 @@ def test_streaming_frontend_emit_session_start_model_none():
     Edge case: Model might be None, should use frontend's model or None.
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         # Frontend with model, but emit_session_start called with None
         frontend = StreamingChunkFrontend(session_id=None, model="claude-sonnet-4.5")
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -518,14 +522,15 @@ def test_streaming_frontend_emit_session_start_multiple_calls():
     Edge case: Multiple calls should work (e.g., if called from different code paths).
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         frontend = StreamingChunkFrontend(session_id=None, model="claude-sonnet-4.5")
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -550,14 +555,15 @@ def test_streaming_frontend_session_start_empty_states():
     Edge case: run_interactive might return empty list (shouldn't crash).
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         frontend = StreamingChunkFrontend(session_id=None, model="claude-sonnet-4.5")
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -583,19 +589,20 @@ def test_streaming_frontend_session_start_state_without_session_id():
     Edge case: First state might not have session_id set yet.
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
     # Mock AgentState for testing
     class MockState:
-        def __init__(self, session_id=None):
+        def __init__(self, session_id=None) -> None:
             self.session_id = session_id
     
-    async def run_test():
+    async def run_test() -> None:
         frontend = StreamingChunkFrontend(session_id=None, model="claude-sonnet-4.5")
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit
@@ -631,9 +638,10 @@ def test_streaming_frontend_session_start_resumed_then_new():
     Edge case: --resume used but states return different session_id (should use states one).
     """
     import trio
+
     from wafer.wevin_cli import StreamingChunkFrontend
     
-    async def run_test():
+    async def run_test() -> None:
         # Start with resumed session_id
         frontend = StreamingChunkFrontend(
             session_id="resumed-session-123",
@@ -641,8 +649,8 @@ def test_streaming_frontend_session_start_resumed_then_new():
         )
         
         emitted_events = []
-        original_emit = frontend._emit
-        def mock_emit(obj):
+
+        def mock_emit(obj) -> None:
             emitted_events.append(obj)
         
         frontend._emit = mock_emit

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/wafer/analytics.py

@@ -15,7 +15,6 @@ Usage:
 
 import atexit
 import platform
-import sys
 import uuid
 from pathlib import Path
 from typing import Any

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/wafer/auth.py

@@ -419,7 +419,7 @@ def device_code_login(timeout: int = 600) -> tuple[str, str | None]:
                     print(f" {CROSS}\n")
                     raise RuntimeError(f"CLI auth flow failed: {response.status_code} {response.text}")
 
-                except httpx.RequestError as e:
+                except httpx.RequestError:
                     # Network error, retry
                     print("!", end="", flush=True)
                     last_poll = time.time()

{wafer_cli-0.2.13 → wafer_cli-0.2.14}/wafer/autotuner.py

@@ -5,6 +5,7 @@ This module provides the implementation for the `wafer autotuner` commands.
 
 import asyncio
 import json
+from datetime import UTC
 from pathlib import Path
 from typing import Any
 
@@ -32,13 +33,14 @@ def run_sweep_command(
         raise FileNotFoundError(f"Config file not found: {config_file}")
 
     # Import autotuner core
-    from datetime import datetime, timezone
+    from datetime import datetime
     from uuid import uuid4
+
     import trio
     from wafer_core.tools.autotuner import AutotunerConfig, run_sweep
     from wafer_core.tools.autotuner.dtypes import Sweep, Trial
     from wafer_core.tools.autotuner.search import generate_grid_trials
-    from wafer_core.tools.autotuner.storage import create_sweep, add_trial, get_sweep, get_trials
+    from wafer_core.tools.autotuner.storage import add_trial, create_sweep, get_sweep, get_trials
 
     # Load or reconstruct config
     if resume_sweep_id:
@@ -189,8 +191,8 @@ def run_sweep_command(
                     status="running",
                     total_trials=total_trials,
                     completed_trials=0,
-                    created_at=datetime.now(timezone.utc),
-                    updated_at=datetime.now(timezone.utc),
+                    created_at=datetime.now(UTC),
+                    updated_at=datetime.now(UTC),
                 )
 
                 # Create sweep and get the actual ID from the API
@@ -245,7 +247,7 @@ def run_sweep_command(
             # Helper to update sweep status
             async def update_sweep_status(status: str) -> None:
                 import httpx
-                from wafer_core.tools.autotuner.storage import get_api_url, _get_auth_headers
+                from wafer_core.tools.autotuner.storage import _get_auth_headers, get_api_url
 
                 api_url = get_api_url()
                 headers = _get_auth_headers()
@@ -260,7 +262,7 @@ def run_sweep_command(
             # Note: working_dir already set based on is_resume flag
 
             try:
-                trials = await run_sweep(
+                await run_sweep(
                     config=config,
                     sweep_id=actual_sweep_id,
                     working_dir=working_dir,
@@ -273,7 +275,7 @@ def run_sweep_command(
 
                 # Print final summary
                 print()
-                print(f"✅ Sweep completed!")
+                print("✅ Sweep completed!")
                 print(f"   Total: {total_trials} trials")
                 print(f"   Success: {success_count}")
                 print(f"   Failed: {failed_count}")
@@ -297,7 +299,7 @@ def run_sweep_command(
             except KeyboardInterrupt:
                 # User pressed Ctrl+C
                 print()
-                print(f"❌ Sweep interrupted by user (Ctrl+C)")
+                print("❌ Sweep interrupted by user (Ctrl+C)")
                 print(f"   Completed: {completed_count}/{total_trials} trials")
                 await update_sweep_status("failed")
                 raise
@@ -350,8 +352,8 @@ def results_command(
         Formatted string with results
     """
     from wafer_core.tools.autotuner import compute_pareto_frontier
-    from wafer_core.tools.autotuner.storage import get_sweep, get_trials
     from wafer_core.tools.autotuner.aggregation import aggregate_trials_by_config
+    from wafer_core.tools.autotuner.storage import get_sweep, get_trials
 
     try:
         # Get sweep and trials
@@ -501,7 +503,10 @@ def results_command(
                     # Use aggregated config scoring
                     if len(objectives_data) > 1:
                         # Multi-objective: compute Pareto
-                        from wafer_core.tools.autotuner.scoring import compute_pareto_frontier_configs, rank_pareto_configs
+                        from wafer_core.tools.autotuner.scoring import (
+                            compute_pareto_frontier_configs,
+                            rank_pareto_configs,
+                        )
                         objectives = [
                             Objective(
                                 metric=obj["metric"],
@@ -513,7 +518,7 @@ def results_command(
                         pareto_configs = compute_pareto_frontier_configs(aggregated_configs, objectives)
                         ranked_configs = rank_pareto_configs(pareto_configs, objectives)
 
-                        lines.append(f"Pareto Frontier (using config objectives):")
+                        lines.append("Pareto Frontier (using config objectives):")
                         lines.append(f"Found {len(ranked_configs)} non-dominated configurations.")
                         lines.append("")
 
@@ -563,7 +568,7 @@ def results_command(
                         ]
                         pareto_trials = compute_pareto_frontier(completed_trials, objectives)
 
-                        lines.append(f"Pareto Frontier (using config objectives):")
+                        lines.append("Pareto Frontier (using config objectives):")
                         lines.append(f"Found {len(pareto_trials)} non-dominated configurations.")
                         lines.append("")
 
@@ -699,8 +704,8 @@ def best_command(
     Returns:
         Formatted string with best config
     """
-    from wafer_core.tools.autotuner.storage import get_sweep, get_trials
     from wafer_core.tools.autotuner.aggregation import aggregate_trials_by_config
+    from wafer_core.tools.autotuner.storage import get_sweep, get_trials
 
     try:
         # Get sweep and trials
@@ -991,7 +996,7 @@ def delete_command(sweep_id: str) -> str:
         Success message
     """
     import httpx
-    from wafer_core.tools.autotuner.storage import get_api_url, _get_auth_headers
+    from wafer_core.tools.autotuner.storage import _get_auth_headers, get_api_url
 
     try:
         api_url = get_api_url()
@@ -1008,8 +1013,7 @@ def delete_command(sweep_id: str) -> str:
     except httpx.HTTPStatusError as e:
         if e.response.status_code == 404:
             raise ValueError(f"Sweep {sweep_id} not found")
-        else:
-            raise ValueError(f"Failed to delete sweep: {e}")
+        raise ValueError(f"Failed to delete sweep: {e}")
     except Exception as e:
         raise ValueError(f"Failed to delete sweep: {e}") from e
 
@@ -1024,7 +1028,7 @@ def delete_all_command(status_filter: str | None = None) -> str:
         Summary of deletions
     """
     import httpx
-    from wafer_core.tools.autotuner.storage import list_sweeps, get_api_url, _get_auth_headers
+    from wafer_core.tools.autotuner.storage import _get_auth_headers, get_api_url, list_sweeps
 
     try:
         # Get all sweeps