No public description

PiperOrigin-RevId: 767522004
Change-Id: I4eed19f4307c2b53024ed354d0d7a87d59571ebe

Author: cahlheim

lightweight_mmm/core/core_utils_test.py

@@ -29,7 +29,7 @@ def test_apply_exponent_safe_produces_same_exponent_results(self):
 
     output = core_utils.apply_exponent_safe(data=data, exponent=exponent)
 
-    np.testing.assert_array_equal(x=output, y=data**exponent)
+    np.testing.assert_array_equal(output, data**exponent)
 
   def test_apply_exponent_safe_produces_correct_shape(self):
     data = jnp.ones((10, 5))

lightweight_mmm/core/time/trend_test.py

@@ -186,8 +186,8 @@ def test_core_dynamic_trend_produces_correct_value(self):
         variance_slope=variance_slope,
     )
 
-    np.testing.assert_array_equal(x=dynamic_trend_values,
-                                  y=dynamic_trend_expected_value)
+    np.testing.assert_array_equal(dynamic_trend_values,
+                                  dynamic_trend_expected_value)
 
   @parameterized.named_parameters([
       dict(

lightweight_mmm/core/transformations/lagging_test.py

@@ -215,7 +215,7 @@ def test_adstock_zeros_stay_zeros(self):
 
     generated_output = lagging._adstock(data=data, lag_weight=lag_weight)
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
   def test_carryover_zeros_stay_zeros(self):
     data = jnp.zeros((10, 5))
@@ -229,7 +229,7 @@ def test_carryover_zeros_stay_zeros(self):
         number_lags=7,
     )
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
 
 if __name__ == "__main__":

lightweight_mmm/core/transformations/saturation_test.py

@@ -188,7 +188,7 @@ def test_hill_zeros_stay_zeros(self):
         slope=slope,
     )
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
   def test_exponent_zeros_stay_zero(self):
     data = jnp.zeros((10, 5))
@@ -199,7 +199,7 @@ def test_exponent_zeros_stay_zero(self):
         exponent_values=exponent_values,
     )
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
 
 if __name__ == "__main__":

lightweight_mmm/media_transforms_test.py

@@ -126,7 +126,7 @@ def test_apply_exponent_safe_produces_correct_shape(self):
 
     output = media_transforms.apply_exponent_safe(data=data, exponent=exponent)
 
-    np.testing.assert_array_equal(x=output, y=data**exponent)
+    np.testing.assert_array_equal(output, data**exponent)
 
   def test_apply_exponent_safe_produces_same_exponent_results(self):
     data = jnp.ones((10, 5))
@@ -156,7 +156,7 @@ def test_adstock_zeros_stay_zeros(self):
     generated_output = media_transforms.adstock(
         data=data, lag_weight=lag_weight)
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
   def test_hill_zeros_stay_zeros(self):
     data = jnp.zeros((10, 5))
@@ -168,7 +168,7 @@ def test_hill_zeros_stay_zeros(self):
         half_max_effective_concentration=half_max_effective_concentration,
         slope=slope)
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
   def test_carryover_zeros_stay_zeros(self):
     data = jnp.zeros((10, 5))
@@ -180,7 +180,7 @@ def test_carryover_zeros_stay_zeros(self):
         ad_effect_retention_rate=ad_effect_retention_rate,
         peak_effect_delay=peak_effect_delay)
 
-    np.testing.assert_array_equal(x=generated_output, y=data)
+    np.testing.assert_array_equal(generated_output, data)
 
 
 @parameterized.parameters(range(1, 5))

requirements/requirements.txt

@@ -18,7 +18,7 @@ immutabledict>=2.0.0
 jax>=0.3.18
 jaxlib>=0.3.18
 matplotlib==3.6.1
-numpy>=1.17.3, <=1.25.2
+numpy>=2.3
 numpyro>=0.9.2
 pandas>=1.1.5
 scipy