code: def test_custom_neg_bin_logpdf_matches_scipy_neg_bin_logpdf(
      total_count, probs
  ):
      torch_dist = NegativeBinomial(
          total_count=total_count, probs=probs
      )
      scipy_dist = torch_dist.scipy_nbinom
      x = torch.randint(1, 20, (1000,))
      log_pdf_torch = torch_dist.log_prob(x).numpy()
      log_pdf_scipy = scipy_dist.logpmf(x.numpy())
  
      assert np.allclose(log_pdf_torch, log_pdf_scipy, rtol=1e-4)
  
imports: import pytest
  import numpy
  import torch
  from gluonts.torch.distributions.negative_binomial import NegativeBinomial
label_desc: Use image node_id 1 for calling a global function with example usage: test_custom_neg_bin_logpdf_matches_scipy_neg_bin_logpdf(total_count, probs) without return types