Adding warnings filter for inference on data sets being empty to avoi…

…d user confusion.
kratzky · Jul 9, 2020 · edac73e · edac73e
1 parent cb78c17
commit edac73e
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 9 deletions.
diff --git a/ImageColorizer.ipynb b/ImageColorizer.ipynb
@@ -21,7 +21,9 @@
    "source": [
     "from deoldify.visualize import *\n",
     "plt.style.use('dark_background')\n",
-    "torch.backends.cudnn.benchmark=True"
+    "torch.backends.cudnn.benchmark=True\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {
@@ -140,7 +142,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.0"
+   "version": "3.7.6"
   },
   "toc": {
    "colors": {

diff --git a/ImageColorizerArtisticTests.ipynb b/ImageColorizerArtisticTests.ipynb
@@ -20,7 +20,9 @@
    "outputs": [],
    "source": [
     "from deoldify.visualize import *\n",
-    "plt.style.use('dark_background')"
+    "plt.style.use('dark_background')\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {
@@ -3285,7 +3287,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.0"
+   "version": "3.7.6"
   },
   "toc": {
    "colors": {

diff --git a/ImageColorizerColab.ipynb b/ImageColorizerColab.ipynb
@@ -144,7 +144,9 @@
    "outputs": [],
    "source": [
     "import fastai\n",
-    "from deoldify.visualize import *"
+    "from deoldify.visualize import *\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {

diff --git a/ImageColorizerStableTests.ipynb b/ImageColorizerStableTests.ipynb
@@ -20,7 +20,9 @@
    "outputs": [],
    "source": [
     "from deoldify.visualize import *\n",
-    "plt.style.use('dark_background')"
+    "plt.style.use('dark_background')\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {

diff --git a/VideoColorizer.ipynb b/VideoColorizer.ipynb
@@ -20,7 +20,9 @@
    "outputs": [],
    "source": [
     "from deoldify.visualize import *\n",
-    "plt.style.use('dark_background')"
+    "plt.style.use('dark_background')\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {
@@ -146,7 +148,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.0"
+   "version": "3.7.6"
   },
   "toc": {
    "colors": {

diff --git a/VideoColorizerColab.ipynb b/VideoColorizerColab.ipynb
@@ -151,7 +151,9 @@
     "import fastai\n",
     "from deoldify.visualize import *\n",
     "from pathlib import Path\n",
-    "torch.backends.cudnn.benchmark=True"
+    "torch.backends.cudnn.benchmark=True\n",
+    "import warnings\n",
+    "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".*?Your .*? set is empty.*?\")"
    ]
   },
   {