Disable highlighting (#23)

* update * add tune setup * update * update * update * update * update * update
fugue-project · Apr 23, 2021 · 615a72e · 615a72e
1 parent 65c21f2
commit 615a72e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -4,6 +4,10 @@ Fugue for Kaggle users
 
 ## Release History
 
+### 0.2.2
+
+* Disable Fugue SQL highlighting on Kaggle, they no longer support it
+
 ### 0.2.1
 
 * Upgrade fugue dependency and add tune dependency and support

diff --git a/fuggle/__init__.py b/fuggle/__init__.py
@@ -4,6 +4,7 @@
 import fugue_notebook as fn
 from fuggle_version import __version__
 from fugue_sql import FugueSQLWorkflow as Dag
+from IPython import get_ipython
 from tune import TUNE_OBJECT_FACTORY, Monitor, NonIterativeObjectiveRunner
 from tune_hyperopt import HyperoptRunner
 from tune_notebook.vis import (
@@ -26,7 +27,9 @@ def setup(default_engine: str = "") -> Any:
     TUNE_OBJECT_FACTORY.set_noniterative_objective_runner_converter(_to_runner)
     TUNE_OBJECT_FACTORY.set_monitor_converter(_to_monitor)
 
-    return fn.setup(KaggleNotebookSetup(default_engine))
+    # we no longer enable SQL highlighting, kaggle has changed
+    ip = get_ipython()
+    fn._setup_fugue_notebook(ip, KaggleNotebookSetup(default_engine))
 
 
 def _to_runner(obj: Any) -> Optional[NonIterativeObjectiveRunner]:

diff --git a/fuggle_version/__init__.py b/fuggle_version/__init__.py
@@ -1 +1 @@
-__version__ = "0.2.1"
+__version__ = "0.2.2"