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()

```

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',
 ]

{"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"}]}