code: def test_timestep_weighted():
      pin_obj = quantiles.PinballLoss()
      y_true = tf.constant([1, 9, 2, -5, -2, 6], shape=(2, 3, 1))
      y_pred = tf.constant(
          [4, 8, 12, 8, 1, 3], shape=(2, 3, 1), dtype=tf.dtypes.float32
      )
      sample_weight = tf.constant([3, 6, 5, 0, 4, 2], shape=(2, 3))
      loss = pin_obj(y_true, y_pred, sample_weight=sample_weight)
      np.testing.assert_almost_equal(loss, 41.5 / 6, 3)
  
file_path: addons/tensorflow_addons/losses/tests/quantiles_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.losses import quantiles
label_desc: Use image node_id 6 for calling a global function with example usage: test_timestep_weighted() without return types