              for gpu in gpus:
                  logger.info("Setting allow growth for GPU: %s", gpu)
                  tf.config.experimental.set_memory_growth(gpu, True)
  
      def load_model(self):
          """ Loads a model.
  
          This method is a wrapper for :func:`keras.models.load_model()`. Loads a model and its
          weights from :attr:`model_path` defined during initialization of this class. Any additional
          ``kwargs`` to be passed to :func:`keras.models.load_model()` should also be defined during
 
</s> remove     def load_model_weights(self):
 </s> add     def load_model_weights(self) -> None: </s> remove     def predict(self, feed, batch_size=None):
 </s> add     def predict(self,
                feed: Union[List[np.ndarray], np.ndarray],
                batch_size: Optional[int] = None) -> Union[List[np.ndarray], np.ndarray]: </s> remove     def append_softmax_activation(self, layer_index=-1):
 </s> add     def append_softmax_activation(self, layer_index: int = -1) -> None: </s> remove         self._model = None
        logger.trace("Initialized: %s", self.__class__.__name__,)
 </s> add         self._model: Optional[Model] = None
        logger.trace("Initialized: %s", self.__class__.__name__,)  # type:ignore
    def load_model(self) -> None:
              for gpu in gpus:
                  logger.info("Setting allow growth for GPU: %s", gpu)
                  tf.config.experimental.set_memory_growth(gpu, True)
  
    def load_model(self) -> None:
          """ Loads a model.
  
          This method is a wrapper for :func:`keras.models.load_model()`. Loads a model and its
          weights from :attr:`model_path` defined during initialization of this class. Any additional
          ``kwargs`` to be passed to :func:`keras.models.load_model()` should also be defined during
keep keep keep keep replace keep keep keep keep keep
 Update TF to 2.9
    - Update TF to 2.7 to 2.9
    - Bump dependencies
    - Remove decode from pynvml calls
    - force keras predict functions to non-verbose
    - update tests
    - update Tensorboard logging
    - Update docs 
lib/model/session.py