fix: add coverage optional extra, fix docs install instructions, reorder init

- Add coverage>=7.0.0 as optional dependency [coverage] extra in pyproject.toml
- Fix docs to reference correct package name tusk-drift-python-sdk[coverage]
- Move start_coverage_collection() after _initialized guard to avoid wasteful re-invocation

Author: sohil-kshirsagar

docs/coverage.md

@@ -5,9 +5,7 @@ The Python SDK collects per-test code coverage during Tusk Drift replay using `c
 ## Requirements
 
 ```bash
-pip install coverage
-# or
-pip install tusk-drift[coverage]
+pip install tusk-drift-python-sdk[coverage]
 ```
 
 If `coverage` is not installed when coverage is enabled, the SDK logs a warning and coverage is skipped. Tests still run normally.

drift/core/drift_sdk.py

@@ -160,12 +160,6 @@ def initialize(
 
         configure_logger(log_level=log_level, prefix="TuskDrift")
 
-        # Start coverage collection early (before any SDK mode checks that might return early).
-        # Coverage data is accessed via protobuf channel (communicator handles requests).
-        from .coverage_server import start_coverage_collection
-
-        start_coverage_collection()
-
         instance._init_params = {
             "api_key": api_key,
             "env": env,
@@ -185,6 +179,13 @@ def initialize(
             )
             env = env_from_var
 
+        # Start coverage collection early (before any SDK mode checks that might return early),
+        # but after the _initialized guard so we don't re-invoke on repeated initialize() calls.
+        # Coverage data is accessed via protobuf channel (communicator handles requests).
+        from .coverage_server import start_coverage_collection
+
+        start_coverage_collection()
+
         if cls._initialized:
             logger.debug("Already initialized, skipping...")
             return instance

pyproject.toml

@@ -44,6 +44,7 @@ dependencies = [
 flask = ["Flask>=3.1.2"]
 fastapi = ["fastapi>=0.115.6", "uvicorn>=0.34.2", "starlette<0.42.0"]
 django = ["Django>=4.2"]
+coverage = ["coverage>=7.0.0"]
 rust = ["drift-core-python>=0.1.9"]
 dev = [
     "Flask>=3.1.2",