code: def get_dataset(self, num_sample):
      X = np.random.randn(num_sample, self.input_dim)
      E = (
          np.random.randn(num_sample, self.output_dim)
          * self.noise_level
      )
      Y = np.matmul(X, self._A) + E
      return SimpleDataset(X, Y)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import pytest
  import numpy
  import torch
  import torch.nn
  import torch.optim
  from torch.utils.data import Dataset, DataLoader
  import bluefog.torch
  from bluefog.common import topology_util
label_desc: Use image node_id 9 for calling the LinearProblemBuilder obj's underlying member method code with example usage: obj.get_dataset(num_sample) and returns: SimpleDataset