Merge branch 'master' into bugfix/tabular_no_categorical

Lightning-Universe · Jan 31, 2022 · 0354b20 · 0354b20
2 parents ee0e740 + 700a9c8
commit 0354b20
Show file tree

Hide file tree

Showing 10 changed files with 49 additions and 0 deletions.
diff --git a/flash/audio/classification/data.py b/flash/audio/classification/data.py
@@ -590,6 +590,8 @@ def from_data_frame(
             >>> import shutil
             >>> shutil.rmtree("train_folder")
             >>> shutil.rmtree("predict_folder")
+            >>> del train_data_frame
+            >>> del predict_data_frame
         """
 
         ds_kw = dict(

diff --git a/flash/image/classification/data.py b/flash/image/classification/data.py
@@ -604,6 +604,8 @@ def from_data_frame(
             >>> import shutil
             >>> shutil.rmtree("train_folder")
             >>> shutil.rmtree("predict_folder")
+            >>> del train_data_frame
+            >>> del predict_data_frame
         """
 
         ds_kw = dict(
@@ -883,6 +885,8 @@ def from_fiftyone(
             >>> import os
             >>> _ = [os.remove(f"image_{i}.png") for i in range(1, 4)]
             >>> _ = [os.remove(f"predict_image_{i}.png") for i in range(1, 4)]
+            >>> del train_dataset
+            >>> del predict_dataset
         """
 
         ds_kw = dict(

diff --git a/flash/tabular/classification/data.py b/flash/tabular/classification/data.py
@@ -150,6 +150,11 @@ def from_data_frame(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
         ds_kw = dict(
             data_pipeline_state=DataPipelineState(),

diff --git a/flash/tabular/forecasting/data.py b/flash/tabular/forecasting/data.py
@@ -162,6 +162,10 @@ def from_data_frame(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del data
         """
 
         ds_kw = dict(

diff --git a/flash/tabular/regression/data.py b/flash/tabular/regression/data.py
@@ -145,6 +145,11 @@ def from_data_frame(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
         ds_kw = dict(
             data_pipeline_state=DataPipelineState(),

diff --git a/flash/text/classification/data.py b/flash/text/classification/data.py
@@ -514,6 +514,11 @@ def from_hf_datasets(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
 
         ds_kw = dict(
@@ -618,6 +623,11 @@ def from_data_frame(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
 
         ds_kw = dict(

diff --git a/flash/text/question_answering/data.py b/flash/text/question_answering/data.py
@@ -770,6 +770,11 @@ def from_dicts(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
 
         ds_kw = dict(

diff --git a/flash/text/seq2seq/summarization/data.py b/flash/text/seq2seq/summarization/data.py
@@ -380,6 +380,11 @@ def from_hf_datasets(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
 
         ds_kw = dict(

diff --git a/flash/text/seq2seq/translation/data.py b/flash/text/seq2seq/translation/data.py
@@ -378,6 +378,11 @@ def from_hf_datasets(
             Training...
             >>> trainer.predict(model, datamodule=datamodule)  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
             Predicting...
+
+        .. testcleanup::
+
+            >>> del train_data
+            >>> del predict_data
         """
 
         ds_kw = dict(

diff --git a/flash/video/classification/data.py b/flash/video/classification/data.py
@@ -517,6 +517,8 @@ def from_data_frame(
             >>> import shutil
             >>> shutil.rmtree("train_folder")
             >>> shutil.rmtree("predict_folder")
+            >>> del train_data_frame
+            >>> del predict_data_frame
         """
 
         ds_kw = dict(
@@ -852,6 +854,8 @@ def from_fiftyone(
             >>> import os
             >>> _ = [os.remove(f"video_{i}.mp4") for i in range(1, 4)]
             >>> _ = [os.remove(f"predict_video_{i}.mp4") for i in range(1, 4)]
+            >>> del train_dataset
+            >>> del predict_dataset
         """
 
         ds_kw = dict(