Adding warnings filter for inference on data sets being empty to avoid user confusion.

jantic · jantic · commit edac73edf1d3 · 2020-07-09T14:32:32.000-07:00
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.*?\")"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,9 @@`
`144`	`144`	`"outputs": [],`
`145`	`145`	`"source": [`
`146`	`146`	`"import fastai\n",`
`147`		`- "from deoldify.visualize import *"`
	`147`	`+ "from deoldify.visualize import *\n",`
	`148`	`+ "import warnings\n",`
	`149`	`+ "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".?Your .? set is empty.*?\")"`
`148`	`150`	`]`
`149`	`151`	`},`
`150`	`152`	`{`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,9 @@`
`20`	`20`	`"outputs": [],`
`21`	`21`	`"source": [`
`22`	`22`	`"from deoldify.visualize import *\n",`
`23`		`- "plt.style.use('dark_background')"`
	`23`	`+ "plt.style.use('dark_background')\n",`
	`24`	`+ "import warnings\n",`
	`25`	`+ "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".?Your .? set is empty.*?\")"`
`24`	`26`	`]`
`25`	`27`	`},`
`26`	`28`	`{`
Original file line number	Diff line number	Diff line change
`@@ -151,7 +151,9 @@`
`151`	`151`	`"import fastai\n",`
`152`	`152`	`"from deoldify.visualize import *\n",`
`153`	`153`	`"from pathlib import Path\n",`
`154`		`- "torch.backends.cudnn.benchmark=True"`
	`154`	`+ "torch.backends.cudnn.benchmark=True\n",`
	`155`	`+ "import warnings\n",`
	`156`	`+ "warnings.filterwarnings(\"ignore\", category=UserWarning, message=\".?Your .? set is empty.*?\")"`
`155`	`157`	`]`
`156`	`158`	`},`
`157`	`159`	`{`