Merge nucleic/sleek-ember-seal-uady into dev

2026-07-30 17:28:44 -07:00
parent 93e9c838bb
commit e4403b570d
7 changed files with 408 additions and 19 deletions
@@ -12,6 +12,16 @@ import train


 class MetricsTests(unittest.TestCase):
+    def test_boundary_training_weight_is_opt_in(self):
+        self.assertEqual(
+            2.0,
+            train.training_weight({"slice": "boundary"}, boundary_weight=2.0),
+        )
+        self.assertEqual(
+            1.0,
+            train.training_weight({"slice": "core"}, boundary_weight=2.0),
+        )
+
    def test_classification_metrics_include_every_label(self):
        actual = list(range(8))
        predicted = [0, 1, 2, 3, 4, 5, 6, 0]