tf-models-nightly 2.20.0.dev20250502__py2.py3-none-any.whl → 2.20.0.dev20250504__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. official/recommendation/uplift/utils.py +2 -2
  2. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/METADATA +1 -1
  3. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/RECORD +7 -7
  4. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/AUTHORS +0 -0
  5. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/LICENSE +0 -0
  6. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/WHEEL +0 -0
  7. {tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/top_level.txt +0 -0
official/recommendation/uplift/utils.py CHANGED
@@ -30,8 +30,8 @@ def expand_to_match_rank(a: tf.Tensor, b: tf.Tensor) -> tf.Tensor:
30
30
  than a, otherwise a `tf.Tensor` of shape (D0, D1, ..., Dn)
31
31
  """
32
32
  rank_deficit = b.shape.rank - a.shape.rank
33
- if rank_deficit > 0:
34
- return tf.reshape(a, a.shape + [1] * rank_deficit)
33
+ for _ in range(rank_deficit):
34
+ a = tf.expand_dims(a, axis=-1)
35
35
  return a
36
36
 
37
37
 
{tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tf-models-nightly
3
- Version: 2.20.0.dev20250502
3
+ Version: 2.20.0.dev20250504
4
4
  Summary: TensorFlow Official Models
5
5
  Home-page: https://github.com/tensorflow/models
6
6
  Author: Google Inc.
{tf_models_nightly-2.20.0.dev20250502.dist-info → tf_models_nightly-2.20.0.dev20250504.dist-info}/RECORD RENAMED
@@ -929,7 +929,7 @@ official/recommendation/uplift/__init__.py,sha256=_jZilTPWKu-MfMaz1IgBjEW6wqkK3F
929
929
  official/recommendation/uplift/keras_test_case.py,sha256=gF5Z2FzXlKAvhuJDdj7PmFj3jsW_ZmUAv_F9Xokvs2M,6156
930
930
  official/recommendation/uplift/keys.py,sha256=7zkxkPIcXceIN5hWm4ATai4h8ymwCj85dU2r8-3XifM,1032
931
931
  official/recommendation/uplift/types.py,sha256=OBpMvU4uAOKXGwN1UIPZw6KWJXBXeuZGT3P3VyDkyWc,4769
932
- official/recommendation/uplift/utils.py,sha256=ZSrzoFJosSdmu7P3yYDo-TER8UQKFimy9ntuXqV4L4c,3501
932
+ official/recommendation/uplift/utils.py,sha256=8XHO5c4DBpjE0Z_z0dilouYBFW4Va45_OK8qPdBWQTk,3490
933
933
  official/recommendation/uplift/utils_test.py,sha256=K-pzTe3MoOEjdy5tLNSRpIQoWWH12MjFTPkhEIHfS3A,4610
934
934
  official/recommendation/uplift/layers/__init__.py,sha256=7l8YBs_4bREV36YgM__8JcpuXU9d-z4eQ8TFxzZpD9s,825
935
935
  official/recommendation/uplift/layers/encoders/__init__.py,sha256=QaQk0BMHJNE8PcaOi72-nkSF5f9_DhDMJtXkTj9xFhk,725
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
1248
1248
  tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
1249
1249
  tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
1250
1250
  tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
1251
- tf_models_nightly-2.20.0.dev20250502.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1252
- tf_models_nightly-2.20.0.dev20250502.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1253
- tf_models_nightly-2.20.0.dev20250502.dist-info/METADATA,sha256=dNBsdiMSsZM_0b0ji43ss2BLmkKBPReYGrBgrOjY1Ow,1432
1254
- tf_models_nightly-2.20.0.dev20250502.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1255
- tf_models_nightly-2.20.0.dev20250502.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1256
- tf_models_nightly-2.20.0.dev20250502.dist-info/RECORD,,
1251
+ tf_models_nightly-2.20.0.dev20250504.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1252
+ tf_models_nightly-2.20.0.dev20250504.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1253
+ tf_models_nightly-2.20.0.dev20250504.dist-info/METADATA,sha256=oR6VtkA5b7azgDFanwJA96NSPlxjN3mDsP6DKwpeiq8,1432
1254
+ tf_models_nightly-2.20.0.dev20250504.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1255
+ tf_models_nightly-2.20.0.dev20250504.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1256
+ tf_models_nightly-2.20.0.dev20250504.dist-info/RECORD,,