code: def _skip_test_for_stateless(self, stateless):
      if not stateless and backend.backend() == "jax":
          self.skipTest(
              "LossScaleOptimizer must use stateless_apply with JAX."
          )
      if stateless and backend.backend() == "tensorflow":
          self.skipTest(
              "stateless_apply is not supported with the TF backend."
          )
  
file_path: keras/keras/optimizers/loss_scale_optimizer_test.py
imports: import numpy
  from absl.testing import parameterized
  from keras import backend
  from keras import ops
  from keras import testing
  from keras.optimizers.loss_scale_optimizer import LossScaleOptimizer
  from keras.optimizers.sgd import SGD
label_desc: Use image node_id 1 for calling the LossScaleOptimizerTest obj's underlying member method code with example usage: obj._skip_test_for_stateless(stateless) without return types