          model_file = next(fname for fname in os.listdir(model_dir) if fname.endswith(".h5"))
          return os.path.join(model_dir, model_file)
  
      def _parse_weights(self,
                         layer: T.Union[keras.models.Model, keras.layers.Layer]) -> dict:
          """ Recursively pass through sub-models to scan layer weights"""
          weights = layer.get_weights()
          logger.debug("Processing weights for layer '%s', length: '%s'",
                       layer.name, len(weights))
  
 
</s> remove                  kernel_size: T.Union[int, T.Tuple[int]],
 </s> add                  kernel_size: int | tuple[int], </s> remove         compiled_layers: T.Dict[str, tf.keras.layers.Layer] = {}
 </s> add         compiled_layers: dict[str, tf.keras.layers.Layer] = {} </s> remove                  kernel_size: T.Union[int, T.Tuple[int, int]] = 3,
 </s> add                  kernel_size: int | tuple[int, int] = 3, </s> remove                             model_name: str) -> Literal[-1, 0, 1]:
 </s> add                             model_name: str) -> T.Literal[-1, 0, 1]: </s> remove     def _get_weights_model(self) -> T.List[keras.models.Model]:
 </s> add     def _get_weights_model(self) -> list[keras.models.Model]:
                       layer: keras.models.Model | keras.layers.Layer) -> dict:
          model_file = next(fname for fname in os.listdir(model_dir) if fname.endswith(".h5"))
          return os.path.join(model_dir, model_file)
  
      def _parse_weights(self,
                       layer: keras.models.Model | keras.layers.Layer) -> dict:
          """ Recursively pass through sub-models to scan layer weights"""
          weights = layer.get_weights()
          logger.debug("Processing weights for layer '%s', length: '%s'",
                       layer.name, len(weights))
  
keep keep keep keep replace keep keep keep keep keep
 Rebase code (#1326)

* Remove tensorflow_probability requirement

* setup.py - fix progress bars

* requirements.txt: Remove pre python 3.9 packages

* update apple requirements.txt

* update INSTALL.md

* Remove python<3.9 code

* setup.py - fix Windows Installer

* typing: python3.9 compliant

* Update pytest and readthedocs python versions

* typing fixes

* Python Version updates
  - Reduce max version to 3.10
  - Default to 3.10 in installers
  - Remove incompatible 3.11 tests

* Update dependencies

* Downgrade imageio dep for Windows

* typing: merge optional unions and fixes

* Updates
  - min python version 3.10
  - typing to python 3.10 spec
  - remove pre-tf2.10 code
  - Add conda tests

* train: re-enable optimizer saving

* Update dockerfiles

* Update setup.py
  - Apple Conda deps to setup.py
  - Better Cuda + dependency handling

* bugfix: Patch logging to prevent Autograph errors

* Update dockerfiles

* Setup.py - Setup.py - stdout to utf-8

* Add more OSes to github Actions

* suppress mac-os end to end test 
tools/model/model.py