tf-models-nightly 2.18.0.dev20241002__py2.py3-none-any.whl → 2.18.0.dev20241004__py2.py3-none-any.whl

official/modeling/multitask/train_lib.py

@@ -40,7 +40,7 @@ def run_experiment(
     *,
     distribution_strategy: tf.distribute.Strategy,
     task: multitask.MultiTask,
-    model: base_model.MultiTaskBaseModel,
+    model: base_model.MultiTaskBaseModel | tf_keras.Model,
     mode: str,
     params: configs.MultiTaskExperimentConfig,
     model_dir: str,

official/projects/movinet/tools/export_saved_model_test.py

@@ -13,13 +13,16 @@
 # limitations under the License.
 
 """Tests for export_saved_model."""
-
 from absl import flags
 import tensorflow as tf, tf_keras
 import tensorflow_hub as hub
 
+# pylint: disable=g-direct-tensorflow-import
+from tensorflow.lite.python import interpreter as tfl_interpreter
+# pylint: enable=g-direct-tensorflow-import
 from official.projects.movinet.tools import export_saved_model
 
+
 FLAGS = flags.FLAGS
 
 
@@ -120,7 +123,7 @@ class ExportSavedModelTest(tf.test.TestCase):
     converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
     tflite_model = converter.convert()
 
-    interpreter = tf.lite.Interpreter(model_content=tflite_model)
+    interpreter = tfl_interpreter.Interpreter(model_content=tflite_model)
     runner = interpreter.get_signature_runner('serving_default')
 
     def state_name(name: str) -> str:

orbit/actions/conditional_action.py

@@ -47,7 +47,8 @@ class ConditionalAction:
     """Initializes the instance.
 
     Args:
-      condition: A callable accepting train or eval outputs and returing a bool.
+      condition: A callable accepting train or eval outputs and returning
+       a bool.
       action: The action (or optionally sequence of actions) to perform when
         `condition` is met.
     """

orbit/actions/new_best_metric.py

@@ -40,7 +40,7 @@ class NewBestMetric:
   if it is achieved. These separate methods enable the same `NewBestMetric`
   instance to be reused as a condition multiple times, and can also provide
   additional preemption/failure safety. For example, to avoid updating the best
-  metric if a model export fails or is pre-empted:
+  metric if a model export fails or is pre-emptied:
 
       new_best_metric = orbit.actions.NewBestMetric(
         'accuracy', filename='/model/dir/best_metric')

orbit/actions/new_best_metric_test.py

@@ -61,7 +61,7 @@ class NewBestMetricTest(tf.test.TestCase):
     tempfile = self.create_tempfile().full_path
     value = {'a': 1, 'b': 2}
     persisted_value = actions.JSONPersistedValue(value, tempfile)
-    # The inital value is used since tempfile is empty.
+    # The initial value is used since tempfile is empty.
     self.assertEqual(persisted_value.read(), value)
     persisted_value = actions.JSONPersistedValue('ignored', tempfile)
     # Initial value of 'ignored' is ignored, since there's a value in tempfile.

{tf_models_nightly-2.18.0.dev20241002.dist-info → tf_models_nightly-2.18.0.dev20241004.dist-info}/METADATA

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

{tf_models_nightly-2.18.0.dev20241002.dist-info → tf_models_nightly-2.18.0.dev20241004.dist-info}/RECORD

@@ -224,7 +224,7 @@ official/modeling/multitask/multitask.py,sha256=DV-ysfhPiIZgsrzZNylsPBxKNBf_xzPx
 official/modeling/multitask/task_sampler.py,sha256=SGVVdjMb5oG4vnCczpfdgBtbsdsXiyBLl9si_0V6nko,4897
 official/modeling/multitask/task_sampler_test.py,sha256=wkPTp1LCNx4uJbfHfVbKrQULQxumZS3ctBemEbnMokk,3037
 official/modeling/multitask/test_utils.py,sha256=fPi_TxtzHy_NGTYmdDAQay9TRjwG70UWHwysDqol4jw,4315
-official/modeling/multitask/train_lib.py,sha256=7Oo2j-EqihKonzkkmjueKNJJzKR3AIl71_pH3gDo7hE,11616
+official/modeling/multitask/train_lib.py,sha256=XmWRj-IOKijIo9EDCRQNRU9rj3qiIDhExiPxDN6mi6A,11633
 official/modeling/multitask/train_lib_test.py,sha256=F_uDEQhSlXq9BH1QBlamUMQ_vRUflli7ef2P9MUKXeg,4766
 official/modeling/optimization/__init__.py,sha256=BI86b89P0xOksBlzOmbAdWOhlJQe6cxtGIRN_zqCwa8,1201
 official/modeling/optimization/adafactor_optimizer.py,sha256=zEHHFg9iH1UwcBSzzYE8bCNWln7QIIftXJEXTGTtCtE,792
@@ -585,7 +585,7 @@ official/projects/movinet/tools/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oR
 official/projects/movinet/tools/convert_3d_2plus1d.py,sha256=1cFp6tswmoXsDPIz3JcSRcTCC3wzsqkI63X2Rh5F4f4,3819
 official/projects/movinet/tools/convert_3d_2plus1d_test.py,sha256=pGoc7vvP3xEbmLHD2uJIZAfM1kYtJl2d9IbuP5IHcBM,2007
 official/projects/movinet/tools/export_saved_model.py,sha256=PxYUKvAyuyZmbyO95A6umon-JO41pl82qTpDaOn8gOk,11704
-official/projects/movinet/tools/export_saved_model_test.py,sha256=SGPtYexVHjxYPP5qLf6QbW_A4X8ayhiuQiGXA3tQ9OU,4461
+official/projects/movinet/tools/export_saved_model_test.py,sha256=M01xLav_jGXDAoan-COLrVG5QLrbqfT51k7fOSCdIps,4624
 official/projects/movinet/tools/quantize_movinet.py,sha256=ihm4SGOiO-YOAq_b8ZOzIjDzJIEAsx2DgR6sAbtHZbI,11994
 official/projects/nhnet/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
 official/projects/nhnet/configs.py,sha256=F5lJNdpd8KZE4PjupSVXffmBlOTTakk_ODKC3eDoaxo,3202
@@ -1195,12 +1195,12 @@ orbit/runner.py,sha256=NrR3p9IGxZpwi05m6dCJndnpaC0TuaLMvMdKMVjYoNw,3519
 orbit/standard_runner.py,sha256=L4503NOVDXS4oqP7YE7Lj-nrOXPB__OzS17fj_KdXPI,17412
 orbit/standard_runner_test.py,sha256=lwiQOvjTpb4HphUuZQ5Hgy2BMbw6lyjEwFy4zYaGKho,5110
 orbit/actions/__init__.py,sha256=Y00W7OONDg3vQoljBx5Vw7b7RJ1mG45GqHCGF-mQyjA,3327
-orbit/actions/conditional_action.py,sha256=F0397S82nxATtXI5CJ9HzqTSore8EJjn4QuGm9dqN6o,1988
+orbit/actions/conditional_action.py,sha256=4bNQoc4GxtpvSsZCSNPkXO8XCTLt3HHdzE9Q--RjCsk,1996
 orbit/actions/conditional_action_test.py,sha256=o3Gd_eR3COFlD-yomiCknL-9VNBIJETosQzEuGMlHSU,1335
 orbit/actions/export_saved_model.py,sha256=QuzV4Nc4hmqEb0ohk244pzowNFbldxgdcoR_Q6evrhw,5960
 orbit/actions/export_saved_model_test.py,sha256=SUxz_WYNh-CIQk7lDvqhxzBgJsodtfuMQhwjDzj676o,10753
-orbit/actions/new_best_metric.py,sha256=UNDnU3P1Ejw1SXKsCd0NyOdVcIR50tRR9zuw3VeiDUU,8489
-orbit/actions/new_best_metric_test.py,sha256=cw9X8qCkCWNjjvUJVKS8KDaVjem5rACtV_6NgJW5NzE,4024
+orbit/actions/new_best_metric.py,sha256=QPsOaL6Cyz32uTNqpJ0pGyI0uk0Uqb3F4ahzm867B9g,8490
+orbit/actions/new_best_metric_test.py,sha256=ir-_ta3tCmPqwTDurgu7fS92K2c87ku_d-R_8oAqonU,4025
 orbit/actions/save_checkpoint_if_preempted.py,sha256=3cDqgRCak2qsmSVwMmcwvGXGkIsajtqX31Kaosz0Rj0,2329
 orbit/examples/__init__.py,sha256=-Npz-uBVln3M-h7s1Ce0ed287ygv7gx5Ivbt8IdI0QE,604
 orbit/examples/single_task/__init__.py,sha256=-Npz-uBVln3M-h7s1Ce0ed287ygv7gx5Ivbt8IdI0QE,604
@@ -1222,9 +1222,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.18.0.dev20241002.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.18.0.dev20241002.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.18.0.dev20241002.dist-info/METADATA,sha256=GoiNhkguuzfHg4wD3a9l_7hXTMsG4HoOFr_KNYm9QJY,1432
-tf_models_nightly-2.18.0.dev20241002.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.18.0.dev20241002.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.18.0.dev20241002.dist-info/RECORD,,
+tf_models_nightly-2.18.0.dev20241004.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.18.0.dev20241004.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.18.0.dev20241004.dist-info/METADATA,sha256=asBzd7ZRlySMX6fMzZLcJXkqTM4-mnYg-V__1ruwCDc,1432
+tf_models_nightly-2.18.0.dev20241004.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.18.0.dev20241004.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.18.0.dev20241004.dist-info/RECORD,,