prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  Export TaskSamplers in __init__
  This will put them in the public API and make them visible in the API docs
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `src/garage/experiment/__init__.py`
  Content:
  ```
  1 """Experiment functions."""
  2 from garage.experiment.local_runner import LocalRunner
  3 from garage.experiment.local_tf_runner import LocalTFRunner
  4 from garage.experiment.meta_evaluator import MetaEvaluator
  5 from garage.experiment.snapshotter import SnapshotConfig, Snapshotter
  6 from garage.experiment.task_sampler import TaskSampler
  7 
  8 __all__ = [
  9     'LocalRunner',
  10     'LocalTFRunner',
  11     'MetaEvaluator',
  12     'Snapshotter',
  13     'SnapshotConfig',
  14     'TaskSampler',
  15 ]
  16
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/src/garage/experiment/__init__.py b/src/garage/experiment/__init__.py
  --- a/src/garage/experiment/__init__.py
  +++ b/src/garage/experiment/__init__.py
  @@ -1,9 +1,15 @@
   """Experiment functions."""
  +# yapf: disable
   from garage.experiment.local_runner import LocalRunner
   from garage.experiment.local_tf_runner import LocalTFRunner
   from garage.experiment.meta_evaluator import MetaEvaluator
   from garage.experiment.snapshotter import SnapshotConfig, Snapshotter
  -from garage.experiment.task_sampler import TaskSampler
  +from garage.experiment.task_sampler import (ConstructEnvsSampler,
  +                                            EnvPoolSampler,
  +                                            SetTaskSampler,
  +                                            TaskSampler)
  +
  +# yapf: enable
   
   __all__ = [
       'LocalRunner',
  @@ -12,4 +18,7 @@
       'Snapshotter',
       'SnapshotConfig',
       'TaskSampler',
  +    'ConstructEnvsSampler',
  +    'EnvPoolSampler',
  +    'SetTaskSampler',
   ]
  
verification_info: {"golden_diff": "diff --git a/src/garage/experiment/__init__.py b/src/garage/experiment/__init__.py\n--- a/src/garage/experiment/__init__.py\n+++ b/src/garage/experiment/__init__.py\n@@ -1,9 +1,15 @@\n \"\"\"Experiment functions.\"\"\"\n+# yapf: disable\n from garage.experiment.local_runner import LocalRunner\n from garage.experiment.local_tf_runner import LocalTFRunner\n from garage.experiment.meta_evaluator import MetaEvaluator\n from garage.experiment.snapshotter import SnapshotConfig, Snapshotter\n-from garage.experiment.task_sampler import TaskSampler\n+from garage.experiment.task_sampler import (ConstructEnvsSampler,\n+                                            EnvPoolSampler,\n+                                            SetTaskSampler,\n+                                            TaskSampler)\n+\n+# yapf: enable\n \n __all__ = [\n     'LocalRunner',\n@@ -12,4 +18,7 @@\n     'Snapshotter',\n     'SnapshotConfig',\n     'TaskSampler',\n+    'ConstructEnvsSampler',\n+    'EnvPoolSampler',\n+    'SetTaskSampler',\n ]\n", "issue": "Export TaskSamplers in __init__\nThis will put them in the public API and make them visible in the API docs\n", "before_files": [{"content": "\"\"\"Experiment functions.\"\"\"\nfrom garage.experiment.local_runner import LocalRunner\nfrom garage.experiment.local_tf_runner import LocalTFRunner\nfrom garage.experiment.meta_evaluator import MetaEvaluator\nfrom garage.experiment.snapshotter import SnapshotConfig, Snapshotter\nfrom garage.experiment.task_sampler import TaskSampler\n\n__all__ = [\n    'LocalRunner',\n    'LocalTFRunner',\n    'MetaEvaluator',\n    'Snapshotter',\n    'SnapshotConfig',\n    'TaskSampler',\n]\n", "path": "src/garage/experiment/__init__.py"}], "after_files": [{"content": "\"\"\"Experiment functions.\"\"\"\n# yapf: disable\nfrom garage.experiment.local_runner import LocalRunner\nfrom garage.experiment.local_tf_runner import LocalTFRunner\nfrom garage.experiment.meta_evaluator import MetaEvaluator\nfrom garage.experiment.snapshotter import SnapshotConfig, Snapshotter\nfrom garage.experiment.task_sampler import (ConstructEnvsSampler,\n                                            EnvPoolSampler,\n                                            SetTaskSampler,\n                                            TaskSampler)\n\n# yapf: enable\n\n__all__ = [\n    'LocalRunner',\n    'LocalTFRunner',\n    'MetaEvaluator',\n    'Snapshotter',\n    'SnapshotConfig',\n    'TaskSampler',\n    'ConstructEnvsSampler',\n    'EnvPoolSampler',\n    'SetTaskSampler',\n]\n", "path": "src/garage/experiment/__init__.py"}]}