Solving lint error

pymc-labs · Oct 13, 2023 · 5fc1b90 · 5fc1b90
1 parent 7953a07
commit 5fc1b90
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/mmm/test_budget_optimizer.py b/tests/mmm/test_budget_optimizer.py
@@ -80,15 +80,15 @@ def test_objective_distribution_invalid_method():
             {"channel1": (0, 50), "channel2": (0, 50)},
             {"channel1": (10, 5), "channel2": (20, 10)},
             ["channel1", "channel2"],
-            {'channel1': 50.0, 'channel2': 50.0},
+            {"channel1": 50.0, "channel2": 50.0},
         ),
         (
             "sigmoid",
             10,
             None,
             {"channel1": (1, 0.5), "channel2": (1, 0.5)},
             ["channel1", "channel2"],
-            {'channel1': 5.0, 'channel2': 5.0},
+            {"channel1": 5.0, "channel2": 5.0},
         ),
     ],
 )
@@ -110,4 +110,4 @@ def test_optimize_budget_distribution_valid(
             )
     # Check if budgets are close to the expected values
     for channel in channels:
-        assert result[channel] == pytest.approx(expected_result[channel], 0.001)
+        assert result[channel] == pytest.approx(expected_result[channel], 0.001)