move base classes to __init__

syftbox/client/base.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING
 
 import httpx
 from loguru import logger
@@ -114,30 +113,3 @@ def from_config(cls, config: SyftClientConfig):
             headers=cls._make_headers(config),
         )
         return cls(conn)
-
-
-@dataclass
-class BaseMetric:
-    """Base class for all metrics with common fields."""
-
-    num_runs: int
-
-
-class MetricCollector(Protocol):
-    """
-    Protocol for classes that collect performance metrics.
-    """
-
-    client_config: SyftClientConfig
-
-    def collect_metrics(self, num_runs: int) -> BaseMetric:
-        """Calculate performance metrics."""
-        ...
-
-
-class BenchmarkReporter(Protocol):
-    """Protocol defining the interface for benchmark result reporters."""
-
-    def generate(self, metrics: dict[str, BaseMetric], report_path: Optional[Path] = None) -> Any:
-        """Generate the benchmark report."""
-        ...

syftbox/client/benchmark/__init__.py

@@ -0,0 +1,34 @@
+from dataclasses import dataclass
+from pathlib import Path
+from typing import TYPE_CHECKING, Any, Optional
+
+from typing_extensions import Protocol
+
+from syftbox.lib.client_config import SyftClientConfig
+
+
+@dataclass
+class BaseMetric:
+    """Base class for all metrics with common fields."""
+
+    num_runs: int
+
+
+class MetricCollector(Protocol):
+    """
+    Protocol for classes that collect performance metrics.
+    """
+
+    client_config: SyftClientConfig
+
+    def collect_metrics(self, num_runs: int) -> BaseMetric:
+        """Calculate performance metrics."""
+        ...
+
+
+class BenchmarkReporter(Protocol):
+    """Protocol defining the interface for benchmark result reporters."""
+
+    def generate(self, metrics: dict[str, BaseMetric], report_path: Optional[Path] = None) -> Any:
+        """Generate the benchmark report."""
+        ...