def __init__()

in ml3/sine_task_sampler.py [0:0]

• 18 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, num_tasks_total, num_tasks_per_batch, num_data_points,
                 input_range=[-5.0, 5.0],
                 amp_range=[1.0, 1.0],
                 freq_range=[1.0, 1.0],
                 phase_range=[np.pi, np.pi],
                 fun_type="sine"):

        self.input_range = input_range

        self.amp_range = amp_range
        self.freq_range = freq_range
        self.phase_range = phase_range

        self.fun_type = fun_type

        self.observation_space = np.ones([1], dtype=np.float32)
        self.action_space = np.ones([1], dtype=np.float32)
        self.sample_space = np.ones([1], dtype=np.float32)

        self.num_tasks_total = num_tasks_total
        self.num_tasks_per_task = num_tasks_per_batch
        self.train_tasks = self._sample_tasks(num_tasks_total, num_data_points)
        self.valid_tasks = self._sample_tasks(num_tasks_total, num_data_points)