              retval = [layer for layer in layers if layer != "keras_encoder"]
              logger.debug("Removing 'keras_encoder' for '%s'", arch)
          return retval
  
      def _get_input_shape(self) -> T.Tuple[int, int, int]:
          """ Obtain the input shape for the model.
  
          Input shape is calculated from the selected Encoder's input size, scaled to the user
          selected Input Scaling, rounded down to the nearest 16 pixels.
  
 
</s> remove     def compute_output_shape(self, input_shape: T.Tuple[int, ...]) -> T.Tuple[int, ...]:
 </s> add     def compute_output_shape(self, input_shape: tuple[int, ...]) -> tuple[int, ...]: </s> remove     def _get_inputs(self) -> T.List[tf.keras.layers.Input]:
 </s> add     def _get_inputs(self) -> list[tf.keras.layers.Input]: </s> remove     def image_shape(self) -> Tuple[int, int, int]:
 </s> add     def image_shape(self) -> tuple[int, int, int]: </s> remove                mode: Literal["full", "cap_max", "cap_min"] = "full") -> T.List[int]:
 </s> add                mode: T.Literal["full", "cap_max", "cap_min"] = "full") -> list[int]: </s> remove     def _mask_shapes(self) -> T.Optional[T.List[tuple]]:
 </s> add     def _mask_shapes(self) -> list[tuple] | None:
    def _get_input_shape(self) -> tuple[int, int, int]:
              retval = [layer for layer in layers if layer != "keras_encoder"]
              logger.debug("Removing 'keras_encoder' for '%s'", arch)
          return retval
  
    def _get_input_shape(self) -> tuple[int, int, int]:
          """ Obtain the input shape for the model.
  
          Input shape is calculated from the selected Encoder's input size, scaled to the user
          selected Input Scaling, rounded down to the nearest 16 pixels.
  
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 
plugins/train/model/phaze_a.py