code: def test_time_dependent_prob_multitask_sampler_const_p():
      sampler = task_sampler.TimeDependentProbMultiTaskSampler(
          task_dict={
              "rte": None,
              "mnli": None,
              "squad_v1": None,
          },
          rng=0,
          task_to_unnormalized_prob_funcs_dict={
              "rte": "1",
              "mnli": "1",
              "squad_v1": "1",
          },
      )
      gold_p = np.ones(3) / 3
      assert np.array_equal(sampler.get_task_p(0), gold_p)
      assert np.array_equal(sampler.get_task_p(500), gold_p)
      assert np.array_equal(sampler.get_task_p(999), gold_p)
  
imports: import numpy
  import pytest
  import jiant.proj.main.components.task_sampler
label_desc: Use image node_id 1 for calling a global function with example usage: test_time_dependent_prob_multitask_sampler_const_p() without return types