margin 0.9.3__tar.gz → 0.9.4__tar.gz

{margin-0.9.3/margin.egg-info → margin-0.9.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: margin
-Version: 0.9.3
+Version: 0.9.4
 Summary: Typed health classification, uncertainty algebra, and correction auditing for any system that measures things and needs to explain what happened.
 Author: Cope Labs LLC
 License: MIT

{margin-0.9.3 → margin-0.9.4}/margin/calibrate.py

@@ -157,10 +157,10 @@ def calibrate_many(
     use_std: bool = False,
     intact_std_multiplier: float = 1.5,
     ablated_std_multiplier: float = 3.0,
-) -> tuple[dict[str, float], dict[str, Thresholds]]:
+    return_parser: bool = False,
+):
     """
-    Calibrate multiple components at once. Returns (baselines, thresholds)
-    ready to pass directly to Parser().
+    Calibrate multiple components at once.
 
     Args:
         component_values:        {name: [healthy_measurements]}
@@ -171,12 +171,14 @@ def calibrate_many(
         use_std:                 if True, use std-based thresholds (see calibrate())
         intact_std_multiplier:   std deviations for intact threshold (std mode)
         ablated_std_multiplier:  std deviations for ablated threshold (std mode)
+        return_parser:           if True, return a ready-to-use Parser instead of the raw tuple
 
     Returns:
-        (baselines_dict, thresholds_dict) suitable for:
+        When return_parser=False (default): (baselines_dict, thresholds_dict) suitable for:
             Parser(baselines=baselines_dict,
                    thresholds=first_threshold,
                    component_thresholds=thresholds_dict)
+        When return_parser=True: a Parser with baselines and per-component thresholds set
     """
     polarities = polarities or {}
     baselines = {}
@@ -193,6 +195,18 @@ def calibrate_many(
         baselines[name] = result.baseline
         thresholds[name] = result.thresholds
 
+    if return_parser:
+        names = list(thresholds.keys())
+        if not names:
+            raise ValueError("No components to calibrate")
+        default_thresholds = thresholds[names[0]]
+        component_thresholds = {n: t for n, t in thresholds.items() if n != names[0]}
+        return Parser(
+            baselines=baselines,
+            thresholds=default_thresholds,
+            component_thresholds=component_thresholds,
+        )
+
     return baselines, thresholds
 
 

{margin-0.9.3 → margin-0.9.4}/margin/streaming.py

@@ -34,6 +34,26 @@ class WindowConfig:
     anomaly: Optional[int] = None
     correlation: Optional[int] = None
 
+    def to_dict(self) -> dict:
+        """Serialize to plain dict; omits None fields."""
+        d: dict = {}
+        if self.drift is not None:
+            d["drift"] = self.drift
+        if self.anomaly is not None:
+            d["anomaly"] = self.anomaly
+        if self.correlation is not None:
+            d["correlation"] = self.correlation
+        return d
+
+    @classmethod
+    def from_dict(cls, d: dict) -> 'WindowConfig':
+        """Deserialize from a plain dict produced by to_dict()."""
+        return cls(
+            drift=d.get("drift"),
+            anomaly=d.get("anomaly"),
+            correlation=d.get("correlation"),
+        )
+
 from .confidence import Confidence
 from .health import Health, Thresholds, classify
 from .observation import Observation, Expression
@@ -478,18 +498,23 @@ class Monitor:
         return self._step
 
     def status(self) -> dict:
-        """Full snapshot: health + drift + anomaly + correlations."""
+        """Full snapshot: health + drift + anomaly + correlations.
+
+        Keys are always present; disabled features or no-data trackers
+        return empty dicts so callers can index without key-existence checks.
+        """
         d: dict = {
             "step": self._step,
             "expression": self._expression.to_dict() if self._expression else None,
+            "drift": {},
+            "anomaly": {},
+            "correlations": {},
         }
         if "drift" in self.features:
-            d["drift"] = {}
             for name, tracker in self._drift_trackers.items():
                 if tracker.classification:
                     d["drift"][name] = tracker.classification.to_dict()
         if "anomaly" in self.features:
-            d["anomaly"] = {}
             for name, tracker in self._anomaly_trackers.items():
                 if tracker.classification:
                     d["anomaly"][name] = tracker.classification.to_dict()

{margin-0.9.3 → margin-0.9.4/margin.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: margin
-Version: 0.9.3
+Version: 0.9.4
 Summary: Typed health classification, uncertainty algebra, and correction auditing for any system that measures things and needs to explain what happened.
 Author: Cope Labs LLC
 License: MIT

{margin-0.9.3 → margin-0.9.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "margin"
-version = "0.9.3"
+version = "0.9.4"
 description = "Typed health classification, uncertainty algebra, and correction auditing for any system that measures things and needs to explain what happened."
 readme = "README.md"
 license = {text = "MIT"}

{margin-0.9.3 → margin-0.9.4}/tests/test_calibrate.py

@@ -176,3 +176,35 @@ class TestNeedsRecalibrationMany:
         flags = needs_recalibration_many(cal, recent)
         assert flags["cpu"] is True
         assert flags["mem"] is False
+
+
+class TestCalibrateManyReturnParser:
+    def test_return_parser_gives_parser(self):
+        from margin.observation import Parser
+        result = calibrate_many(
+            {"rps": [90.0, 100.0, 110.0], "latency": [48.0, 50.0, 52.0]},
+            polarities={"latency": False},
+            return_parser=True,
+        )
+        assert isinstance(result, Parser)
+
+    def test_return_parser_classifies_correctly(self):
+        result = calibrate_many(
+            {"rps": [100.0] * 10, "latency": [50.0] * 10},
+            polarities={"latency": False},
+            return_parser=True,
+        )
+        from margin.health import Health
+        expr = result.parse({"rps": 95.0, "latency": 52.0})
+        assert expr.health_of("rps") == Health.INTACT
+        assert expr.health_of("latency") == Health.INTACT
+
+    def test_return_parser_false_gives_tuple(self):
+        result = calibrate_many({"rps": [100.0] * 5})
+        assert isinstance(result, tuple)
+        assert len(result) == 2
+
+    def test_return_parser_empty_raises(self):
+        import pytest
+        with pytest.raises(ValueError):
+            calibrate_many({}, return_parser=True)

{margin-0.9.3 → margin-0.9.4}/tests/test_streaming.py

@@ -6,6 +6,7 @@ from margin import (
     Observation, Health, Confidence, Thresholds, Parser,
     DriftState, DriftDirection, AnomalyState,
     DriftTracker, AnomalyTracker, CorrelationTracker, Monitor,
+    WindowConfig,
 )
 
 
@@ -330,15 +331,16 @@ class TestMonitorFeatures:
         m = Monitor(self._parser(), features={"health", "drift"})
         assert isinstance(m.features, frozenset)
 
-    def test_status_omits_disabled_feature_keys(self):
+    def test_status_always_has_drift_anomaly_correlations_keys(self):
         m = Monitor(self._parser(), features={"health"})
         for i in range(5):
             m.update({"cpu": 50.0 - i, "mem": 70.0},
                      now=t0 + timedelta(seconds=i * 60))
         s = m.status()
-        # drift disabled → "drift" key absent
-        assert "drift" not in s
-        assert "anomaly" not in s
+        # Keys always present — empty dict when feature disabled or no data yet
+        assert "drift" in s and s["drift"] == {}
+        assert "anomaly" in s and s["anomaly"] == {}
+        assert "correlations" in s and s["correlations"] == {}
 
     def test_repr_shows_features_when_partial(self):
         m = Monitor(self._parser(), features={"health", "drift"})
@@ -405,3 +407,44 @@ class TestMonitorResetAnomalyReference:
         m = Monitor(self._parser(), features={"health", "drift"})
         # Should not raise even though anomaly trackers don't exist
         m.reset_anomaly_reference()
+
+
+# -----------------------------------------------------------------------
+# WindowConfig serialization
+# -----------------------------------------------------------------------
+
+class TestWindowConfigSerialization:
+    def test_to_dict_full(self):
+        wc = WindowConfig(drift=10, anomaly=40, correlation=100)
+        d = wc.to_dict()
+        assert d == {"drift": 10, "anomaly": 40, "correlation": 100}
+
+    def test_to_dict_partial(self):
+        wc = WindowConfig(drift=10)
+        d = wc.to_dict()
+        assert d == {"drift": 10}
+        assert "anomaly" not in d
+        assert "correlation" not in d
+
+    def test_to_dict_empty(self):
+        wc = WindowConfig()
+        assert wc.to_dict() == {}
+
+    def test_from_dict_roundtrip(self):
+        wc = WindowConfig(drift=10, anomaly=40, correlation=100)
+        wc2 = WindowConfig.from_dict(wc.to_dict())
+        assert wc2.drift == 10
+        assert wc2.anomaly == 40
+        assert wc2.correlation == 100
+
+    def test_from_dict_partial(self):
+        wc = WindowConfig.from_dict({"drift": 20})
+        assert wc.drift == 20
+        assert wc.anomaly is None
+        assert wc.correlation is None
+
+    def test_from_dict_empty(self):
+        wc = WindowConfig.from_dict({})
+        assert wc.drift is None
+        assert wc.anomaly is None
+        assert wc.correlation is None