                                 model_name="squeezenet_lpips_v1.h5"),
              "vgg16": NetInfo(model_id=20,
                               model_name="vgg16_lpips_v1.h5")}
  
      def _linear_block(self, net_output_layer: tf.Tensor) -> Tuple[tf.Tensor, tf.Tensor]:
          """ Build a linear block for a trunk network output.
  
          Parameters
          ----------
          net_output_layer: :class:`tensorflow.Tensor`
 
</s> remove     def _get_ssim(self, y_true: tf.Tensor, y_pred: tf.Tensor) -> Tuple[tf.Tensor, tf.Tensor]:
 </s> add     def _get_ssim(self, y_true: tf.Tensor, y_pred: tf.Tensor) -> tuple[tf.Tensor, tf.Tensor]: </s> remove                     mask_prop: float = 1.0) -> Tuple[tf.Tensor, tf.Tensor]:
 </s> add                     mask_prop: float = 1.0) -> tuple[tf.Tensor, tf.Tensor]: </s> remove     def _get_laplacian_pyramid(self, inputs: tf.Tensor) -> List[tf.Tensor]:
 </s> add     def _get_laplacian_pyramid(self, inputs: tf.Tensor) -> list[tf.Tensor]: </s> remove     def _get_function(self, name: str) -> T.Callable[[tf.Tensor, tf.Tensor], tf.Tensor]:
 </s> add     def _get_function(self, name: str) -> Callable[[tf.Tensor, tf.Tensor], tf.Tensor]: </s> remove     def __call__(self, images: np.ndarray) -> T.List[np.ndarray]:
 </s> add     def __call__(self, images: np.ndarray) -> list[np.ndarray]:
    def _linear_block(self, net_output_layer: tf.Tensor) -> tuple[tf.Tensor, tf.Tensor]:
                                 model_name="squeezenet_lpips_v1.h5"),
              "vgg16": NetInfo(model_id=20,
                               model_name="vgg16_lpips_v1.h5")}
  
    def _linear_block(self, net_output_layer: tf.Tensor) -> tuple[tf.Tensor, tf.Tensor]:
          """ Build a linear block for a trunk network output.
  
          Parameters
          ----------
          net_output_layer: :class:`tensorflow.Tensor`
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 
lib/model/losses/feature_loss.py