  @pytest.mark.parametrize('dummy', [None], ids=[get_backend().upper()])
  def test_keras(dummy):  # pylint:disable=unused-argument
      """ Sanity check to ensure that tensorflow keras is being used for CPU and standard
      keras for AMD. """
      assert ((_BACKEND == "cpu" and keras.__version__ in ("2.4.0", "2.6.0", "2.7.0", "2.8.0")) or
              (_BACKEND == "amd" and keras.__version__ == "2.2.4"))
 
</s> remove     def _setup_amd(cls, arguments):
 </s> add     def _setup_amd(cls, arguments: "argparse.Namespace") -> bool: </s> remove from lib.utils import get_backend, get_tf_version
 </s> add from lib.utils import get_backend </s> remove         standard = self._model.model.predict([feed_a, feed_b])
        swapped = self._model.model.predict([feed_b, feed_a])
 </s> add         standard = self._model.model.predict([feed_a, feed_b], verbose=0)
        swapped = self._model.model.predict([feed_b, feed_a], verbose=0)
    assert ((_BACKEND == "cpu" and keras.__version__ in ("2.7.0", "2.8.0", "2.9.0")) or
  @pytest.mark.parametrize('dummy', [None], ids=[get_backend().upper()])
  def test_keras(dummy):  # pylint:disable=unused-argument
      """ Sanity check to ensure that tensorflow keras is being used for CPU and standard
      keras for AMD. """
    assert ((_BACKEND == "cpu" and keras.__version__ in ("2.7.0", "2.8.0", "2.9.0")) or
              (_BACKEND == "amd" and keras.__version__ == "2.2.4"))
keep keep keep keep replace 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 
tests/startup_test.py