add random datasets (#266)

* enabled manual returns * enabled manual returns
Lightning-Universe · Oct 12, 2020 · 7b1d895 · 7b1d895
1 parent eac5839
commit 7b1d895
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 78 deletions.
diff --git a/docs/source/dataloaders.rst b/docs/source/dataloaders.rst
@@ -14,11 +14,3 @@ Example:
 
 .. autoclass:: pl_bolts.datamodules.async_dataloader.AsynchronousLoader
  :noindex:
-
-------------------
-
-DummyDataset
-------------
-
-.. autoclass:: pl_bolts.datamodules.dummy_dataset.DummyDataset
- :noindex:
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -33,6 +33,13 @@ PyTorch-Lightning-Bolts documentation
  sklearn_datamodule
  vision_datamodules
 
+.. toctree::
+ :maxdepth: 2
+ :name: datasets
+ :caption: Datasets
+
+ datasets
+
 .. toctree::
  :maxdepth: 2
  :name: dataloaders
@@ -53,8 +60,14 @@ PyTorch-Lightning-Bolts documentation
  :caption: Models
 
  models_howto
- autoencoders
  classic_ml
+
+.. toctree::
+ :maxdepth: 2
+ :name: vision
+ :caption: Vision models
+
+ autoencoders
  convolutional
  gans
  reinforce_learn
@@ -91,6 +104,7 @@ Indices and tables
  readme
  api/pl_bolts.callbacks
  api/pl_bolts.datamodules
+ api/pl_bolts.datasets
  api/pl_bolts.metrics
  api/pl_bolts.models
  api/pl_bolts.callbacks

diff --git a/pl_bolts/datamodules/__init__.py b/pl_bolts/datamodules/__init__.py
@@ -1,5 +1,4 @@
 from pl_bolts.datamodules.async_dataloader import AsynchronousLoader
-from pl_bolts.datamodules.dummy_dataset import DummyDataset, DummyDetectionDataset
 
 try:
  from pl_bolts.datamodules.binary_mnist_datamodule import BinaryMNISTDataModule

diff --git a/pl_bolts/datamodules/dummy_dataset.py b/pl_bolts/datamodules/dummy_dataset.py