tf-models-nightly 2.20.0.dev20250501__py2.py3-none-any.whl → 2.20.0.dev20250503__py2.py3-none-any.whl

official/projects/centernet/modeling/heads/centernet_head.py

@@ -102,4 +102,4 @@ class CenterNetHead(tf_keras.Model):
   @property
   def output_specs(self) -> Mapping[str, tf.TensorShape]:
     """A dict of {level: TensorShape} pairs for the model output."""
-    return self._output_specs
+    return self._output_specs  # pytype: disable=bad-return-type

official/recommendation/uplift/utils.py

@@ -30,8 +30,8 @@ def expand_to_match_rank(a: tf.Tensor, b: tf.Tensor) -> tf.Tensor:
     than a, otherwise a `tf.Tensor` of shape (D0, D1, ..., Dn)
   """
   rank_deficit = b.shape.rank - a.shape.rank
-  if rank_deficit > 0:
-    return tf.reshape(a, a.shape + [1] * rank_deficit)
+  for _ in range(rank_deficit):
+    a = tf.expand_dims(a, axis=-1)
   return a
 
 

{tf_models_nightly-2.20.0.dev20250501.dist-info → tf_models_nightly-2.20.0.dev20250503.dist-info}/METADATA

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

{tf_models_nightly-2.20.0.dev20250501.dist-info → tf_models_nightly-2.20.0.dev20250503.dist-info}/RECORD

@@ -487,7 +487,7 @@ official/projects/centernet/modeling/backbones/__init__.py,sha256=7oiypy0N82PDw9
 official/projects/centernet/modeling/backbones/hourglass.py,sha256=Nqie343NVOrGKq7kIDrx5rwQ74-lCxW-5IxXUz3ghxQ,10295
 official/projects/centernet/modeling/backbones/hourglass_test.py,sha256=rg8XnC9HNaZEtEj_LZ2jckQP5gIOVbsyBVQdtfGmeV0,1612
 official/projects/centernet/modeling/heads/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
-official/projects/centernet/modeling/heads/centernet_head.py,sha256=QVdL9PR0c3DoYOOXhsPJTfhj4uQm-jjvWusiqweDw1g,3843
+official/projects/centernet/modeling/heads/centernet_head.py,sha256=FMiniiG-kEzh_iQxbIigyhf1IK_dmiQINex27QmwFxs,3878
 official/projects/centernet/modeling/heads/centernet_head_test.py,sha256=TH6hESk-_TPdcqNlwCOPsdTKD2XYSyECyezuLH0AIbw,2435
 official/projects/centernet/modeling/layers/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
 official/projects/centernet/modeling/layers/cn_nn_blocks.py,sha256=pP0vxJMSS233eyquVe9W4STKfPm3Arnt-4T52tMKI9w,12210
@@ -929,7 +929,7 @@ official/recommendation/uplift/__init__.py,sha256=_jZilTPWKu-MfMaz1IgBjEW6wqkK3F
 official/recommendation/uplift/keras_test_case.py,sha256=gF5Z2FzXlKAvhuJDdj7PmFj3jsW_ZmUAv_F9Xokvs2M,6156
 official/recommendation/uplift/keys.py,sha256=7zkxkPIcXceIN5hWm4ATai4h8ymwCj85dU2r8-3XifM,1032
 official/recommendation/uplift/types.py,sha256=OBpMvU4uAOKXGwN1UIPZw6KWJXBXeuZGT3P3VyDkyWc,4769
-official/recommendation/uplift/utils.py,sha256=ZSrzoFJosSdmu7P3yYDo-TER8UQKFimy9ntuXqV4L4c,3501
+official/recommendation/uplift/utils.py,sha256=8XHO5c4DBpjE0Z_z0dilouYBFW4Va45_OK8qPdBWQTk,3490
 official/recommendation/uplift/utils_test.py,sha256=K-pzTe3MoOEjdy5tLNSRpIQoWWH12MjFTPkhEIHfS3A,4610
 official/recommendation/uplift/layers/__init__.py,sha256=7l8YBs_4bREV36YgM__8JcpuXU9d-z4eQ8TFxzZpD9s,825
 official/recommendation/uplift/layers/encoders/__init__.py,sha256=QaQk0BMHJNE8PcaOi72-nkSF5f9_DhDMJtXkTj9xFhk,725
@@ -1248,9 +1248,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.20.0.dev20250501.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.20.0.dev20250501.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.20.0.dev20250501.dist-info/METADATA,sha256=UdBY-epZ_leUpsANovREBKaSFSFkIIriZys1P7pFV9Q,1432
-tf_models_nightly-2.20.0.dev20250501.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.20.0.dev20250501.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.20.0.dev20250501.dist-info/RECORD,,
+tf_models_nightly-2.20.0.dev20250503.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.20.0.dev20250503.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.20.0.dev20250503.dist-info/METADATA,sha256=V37P85_Jv6f1zTN7SowyrBC3Q1qLzmgm60wiGKHtyhw,1432
+tf_models_nightly-2.20.0.dev20250503.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.20.0.dev20250503.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.20.0.dev20250503.dist-info/RECORD,,