Fix IPython error display (nerfstudio-project#1397)

machenmusik · Feb 10, 2023 · 259f3aa · 259f3aa
1 parent 3cf2bb0
commit 259f3aa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/colab/demo.ipynb b/colab/demo.ipynb
@@ -292,9 +292,9 @@
         "if os.path.exists(f\"data/nerfstudio/{scene}/transforms.json\"):\n",
         "    !ns-train nerfacto --viewer.websocket-port 7007 nerfstudio-data --data data/nerfstudio/$scene --downscale-factor 4\n",
         "else:\n",
-        "    import IPython\n",
-        "    display(IPython.display.HTML('<h3 style=\"color:red\">Error: Data processing did not complete</h3>'))\n",
-        "    display(IPython.display.HTML('<h3>Please re-run `Downloading and Processing Data`, or view the FAQ for more info.</h3>'))"
+        "    from IPython.core.display import display, HTML\n",
+        "    display(HTML('<h3 style=\"color:red\">Error: Data processing did not complete</h3>'))\n",
+        "    display(HTML('<h3>Please re-run `Downloading and Processing Data`, or view the FAQ for more info.</h3>'))"
       ]
     },
     {
@@ -383,4 +383,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}