code: def get_test_input_logits_only_with_sample_weights(n_train, n_test):
      """Get example input logits for attacks."""
      rng = np.random.RandomState(4)
      return AttackInputData(
          logits_train=rng.randn(n_train, 5) + 0.2,
          logits_test=rng.randn(n_test, 5) + 0.2,
          sample_weight_train=rng.randn(n_train, 1),
          sample_weight_test=rng.randn(n_test, 1),
      )
  
imports: from unittest import mock
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from tensorflow_privacy.privacy.privacy_tests import epsilon_lower_bound
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import membership_inference_attack
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import models
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import AttackInputData
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import AttackType
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import DataSize
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SingleSliceSpec
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SlicingFeature
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SlicingSpec
label_desc: Use image node_id 6 for calling a global function with example usage: get_test_input_logits_only_with_sample_weights(n_train, n_test) and returns: AttackInputData