tf-models-nightly 2.17.0.dev20240617__py2.py3-none-any.whl → 2.17.0.dev20240618__py2.py3-none-any.whl

official/recommendation/uplift/metrics/variance_test.py

@@ -138,19 +138,19 @@ class VarianceTest(keras_test_case.KerasTestCase, parameterized.TestCase):
   def test_float_sample_weight(self, values, sample_weight, expected_variance):
     metric = variance.Variance()
     metric(values, sample_weight=sample_weight)
-    self.assertEqual(expected_variance, metric.result())
+    self.assertAllClose(expected_variance, metric.result())
 
   def test_empty_input(self):
     metric = variance.Variance()
     values = tf.constant([0, 1, 2, 3])
     metric(values)
-    self.assertEqual(1.25, metric.result())
+    self.assertAllClose(1.25, metric.result())
     metric(tf.ones(shape=(0,)), sample_weight=None)
-    self.assertEqual(1.25, metric.result())
+    self.assertAllClose(1.25, metric.result())
 
   def test_initial_state(self):
     metric = variance.Variance()
-    self.assertEqual(0.0, metric.result())
+    self.assertAllClose(0.0, metric.result())
 
   def test_dtype_correctness(self):
     # 1 << 128 overflows for float32 but fits in float64.
@@ -196,24 +196,26 @@ class VarianceTest(keras_test_case.KerasTestCase, parameterized.TestCase):
     values = tf.constant([1, 2, 1, 4])
     metric.update_state(values)
 
-    self.assertEqual(values.numpy().var(), metric.result())
-    self.assertEqual(values.numpy().var(), metric.result())
+    self.assertAllClose(values.numpy().var(), metric.result())
+    self.assertAllClose(values.numpy().var(), metric.result())
 
     metric.update_state(tf.constant([-1, -2, 0]))
 
-    self.assertEqual(np.array([1, 2, 1, 4, -1, -2, 0]).var(), metric.result())
+    self.assertAllClose(
+        np.array([1, 2, 1, 4, -1, -2, 0]).var(), metric.result()
+    )
 
   def test_reset_state(self):
     metric = variance.Variance()
     values = tf.constant([1, 2, 1, 4])
 
     metric.update_state(values)
-    self.assertEqual(1.5, metric.result())
+    self.assertAllClose(1.5, metric.result())
 
     metric.reset_state()
 
     metric.update_state(values, sample_weight=tf.constant([1, 0, 1, 0]))
-    self.assertEqual(0.0, metric.result())
+    self.assertAllClose(0.0, metric.result())
 
   def test_numpy_correctness(self):
     metric = variance.Variance()

{tf_models_nightly-2.17.0.dev20240617.dist-info → tf_models_nightly-2.17.0.dev20240618.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.17.0.dev20240617
+Version: 2.17.0.dev20240618
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.17.0.dev20240617.dist-info → tf_models_nightly-2.17.0.dev20240618.dist-info}/RECORD

@@ -930,7 +930,7 @@ official/recommendation/uplift/metrics/treatment_sliced_metric_test.py,sha256=-S
 official/recommendation/uplift/metrics/uplift_mean.py,sha256=9I2_8p3bXj5S5MF1pcKzb761bE9lF45OOuInFgEK6HQ,3425
 official/recommendation/uplift/metrics/uplift_mean_test.py,sha256=oZZjdSORNGhQLUMmUy-imz65Kc3OapzKFH5iYYdMmS4,7411
 official/recommendation/uplift/metrics/variance.py,sha256=rhwZzUX-cRbwr-7vhC0I0bEQ9KpeHoPq2T0tijdO3G4,2332
-official/recommendation/uplift/metrics/variance_test.py,sha256=EPISeHOFIh6WfODuC0SXbnmMugh90acMmm4BJkEZXlo,7757
+official/recommendation/uplift/metrics/variance_test.py,sha256=rSKjdAGxCoZNL3Y_cMFjp7fzWNoRMl_uRFzW5ddpNOs,7798
 official/recommendation/uplift/models/__init__.py,sha256=kWy2K5LGXHVyyrTjJvbVFcBjj1bjPRI2dpIq-sfdhvo,716
 official/recommendation/uplift/models/two_tower_uplift_model.py,sha256=Fb6nLFAOqch81ravK57K9kggAeqvtJcBtKGZwCex0ts,5028
 official/recommendation/uplift/models/two_tower_uplift_model_test.py,sha256=J7qC9f0fDG1aIrLz85K1qUzTFyAIH0v8eA1yfPJb9YY,10061
@@ -1212,9 +1212,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
 tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
 tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
 tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
-tf_models_nightly-2.17.0.dev20240617.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.17.0.dev20240617.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.17.0.dev20240617.dist-info/METADATA,sha256=smo4EnD9zKiTmnXASYFdhNZWvd7hPuGFZoCZZx39UP8,1432
-tf_models_nightly-2.17.0.dev20240617.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.17.0.dev20240617.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.17.0.dev20240617.dist-info/RECORD,,
+tf_models_nightly-2.17.0.dev20240618.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.17.0.dev20240618.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.17.0.dev20240618.dist-info/METADATA,sha256=8e2zoMsfMjS1K7i_0fiD5ryok1ZSBwb42dwZbHgLUIc,1432
+tf_models_nightly-2.17.0.dev20240618.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.17.0.dev20240618.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.17.0.dev20240618.dist-info/RECORD,,