tf-models-nightly 2.18.0.dev20241029__py2.py3-none-any.whl → 2.19.0.dev20241030__py2.py3-none-any.whl

official/modeling/hyperparams/base_config.py

@@ -18,6 +18,7 @@ import copy
 import dataclasses
 import functools
 import inspect
+import types
 import typing
 from typing import Any, List, Mapping, Optional, Type, Union
 
@@ -58,8 +59,13 @@ def bind(config_cls):
 
 
 def _is_optional(field):
-  return typing.get_origin(field) is Union and type(None) in typing.get_args(
-      field)
+  # Two styles of annotating optional fields:
+  #   Optional[T] -> typing.Union
+  #   T | None    -> types.UnionType
+  is_union = typing.get_origin(field) in (Union, types.UnionType)
+  # An optional field is a union of a type, and NoneType.
+  args = typing.get_args(field)
+  return is_union and len(args) == 2 and type(None) in args
 
 
 @dataclasses.dataclass
@@ -189,6 +195,9 @@ class Config(params_dict.ParamsDict):
     if not subconfig_type:
       subconfig_type = Config
 
+    def _is_subtype(x, target) -> bool:
+      return isinstance(x, type) and issubclass(x, target)
+
     annotations = cls._get_annotations()
     if k in annotations:
       # Directly Config subtype.
@@ -198,22 +207,16 @@ class Config(params_dict.ParamsDict):
       traverse_in = True
       while traverse_in:
         i += 1
-        if (isinstance(type_annotation, type) and
-            issubclass(type_annotation, Config)):
+        if _is_subtype(type_annotation, Config):
           subconfig_type = type_annotation
           break
         else:
-          # Check if the field is a sequence of subtypes.
-          field_type = typing.get_origin(type_annotation)
-          if (isinstance(field_type, type) and
-              issubclass(field_type, cls.SEQUENCE_TYPES)):
-            element_type = typing.get_args(type_annotation)[0]
-            subconfig_type = (
-                element_type if issubclass(element_type, params_dict.ParamsDict)
-                else subconfig_type)
-            break
-          elif _is_optional(type_annotation):
-            # Strip the `Optional` annotation and process the subtype.
+          # If the field is a sequence of sub-config types or an Optional
+          # sub-config, then strip the container and process the sub-config.
+          is_sequence = _is_subtype(
+              typing.get_origin(type_annotation), cls.SEQUENCE_TYPES
+          )
+          if is_sequence or _is_optional(type_annotation):
             type_annotation = typing.get_args(type_annotation)[0]
             continue
         traverse_in = False

official/modeling/hyperparams/base_config_test.py

@@ -61,6 +61,12 @@ class DumpConfig6(base_config.Config):
   test_config1: Optional[DumpConfig1] = None
 
 
+@dataclasses.dataclass
+class ModernOptionalConfig(base_config.Config):
+  leaf: DumpConfig1 | None = None
+  leaves: tuple[DumpConfig1 | None, ...] = tuple()
+
+
 class BaseConfigTest(parameterized.TestCase, tf.test.TestCase):
 
   def assertHasSameTypes(self, c, d, msg=''):
@@ -422,6 +428,20 @@ class BaseConfigTest(parameterized.TestCase, tf.test.TestCase):
                      "DumpConfig6(test_config1=DumpConfig1(a=1, b='abc'))")
     self.assertIsInstance(c.test_config1, DumpConfig1)
 
+  def test_modern_optional_syntax(self):
+    config = ModernOptionalConfig()
+    self.assertIsNone(config.leaf)
+    self.assertEqual(config.leaves, tuple())
+
+    replaced = config.replace(leaf={'a': 2}, leaves=({'a': 3}, {'b': 'foo'}))
+    self.assertEqual(replaced.leaf.a, 2)
+    self.assertEqual(replaced.leaf.b, 'text')
+    self.assertLen(replaced.leaves, 2)
+    self.assertEqual(replaced.leaves[0].a, 3)
+    self.assertEqual(replaced.leaves[0].b, 'text')
+    self.assertEqual(replaced.leaves[1].a, 1)
+    self.assertEqual(replaced.leaves[1].b, 'foo')
+
 
 if __name__ == '__main__':
   tf.test.main()

{tf_models_nightly-2.18.0.dev20241029.dist-info → tf_models_nightly-2.19.0.dev20241030.dist-info}/METADATA

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

{tf_models_nightly-2.18.0.dev20241029.dist-info → tf_models_nightly-2.19.0.dev20241030.dist-info}/RECORD

@@ -205,8 +205,8 @@ official/modeling/activations/sigmoid_test.py,sha256=BcG8HgPHcddhBHyqIO2xVvw1sBz
 official/modeling/activations/swish.py,sha256=1T7WlQJ2plHk3JWyBiHz-BFgnkWcwudI_UU_hu6N9UQ,2301
 official/modeling/activations/swish_test.py,sha256=7cfqupqDQ1_b0F7y7tlXKcqUP4tN7LBxgn8Hjzf4a18,1421
 official/modeling/hyperparams/__init__.py,sha256=vs543Hge6bXJVYMNcsLBLSv-AvkqiC5FbRziP5gD3uA,846
-official/modeling/hyperparams/base_config.py,sha256=F7D8EqMI3DOC6SR4XfJmuEiRgrBRXbJ7ETJ-E0WaeDc,12519
-official/modeling/hyperparams/base_config_test.py,sha256=Jwx0O1iHUtQ8wj1ZO8xQ2wRpnU9AyH8W43cwHHFCsqc,12636
+official/modeling/hyperparams/base_config.py,sha256=mHwYaOj4hwyhlZSh2pxrdJf4-xg00lW-nF_t-LYCKH0,12582
+official/modeling/hyperparams/base_config_test.py,sha256=eAg7tcnFDiMINA84LutiPGVqRRDpG4JGSc5mzn_MOxM,13350
 official/modeling/hyperparams/oneof.py,sha256=_fRv3YICiAru-u5q2YT0QEGA1-t8LBsKGAwfN3jwbA0,1870
 official/modeling/hyperparams/oneof_test.py,sha256=obwub_gkT4El0U5O58damISqVHKF-i5HBeyGA5ot74s,1991
 official/modeling/hyperparams/params_dict.py,sha256=63fftQdUlycgJErxcyIj7655zL57isskkwLcUP41lfY,17810
@@ -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.dev20241029.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.18.0.dev20241029.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.18.0.dev20241029.dist-info/METADATA,sha256=HEhluJ8CbB8Va_H4heir1YVyGUHttLMRMhyAFUlN9oo,1432
-tf_models_nightly-2.18.0.dev20241029.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.18.0.dev20241029.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.18.0.dev20241029.dist-info/RECORD,,
+tf_models_nightly-2.19.0.dev20241030.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.19.0.dev20241030.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.19.0.dev20241030.dist-info/METADATA,sha256=TNjxVJQXfk3-KT7PiqLROuGQMAOoZ8Dg9YtGwkebnWg,1432
+tf_models_nightly-2.19.0.dev20241030.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.19.0.dev20241030.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.19.0.dev20241030.dist-info/RECORD,,