HiPlot in Ax

Reviewed By: 2timesjay

Differential Revision: D21302382

fbshipit-source-id: 082db90c6674e1049c4bba808536c8b65b339027

Author: danthe3rd

ax/plot/base.py

@@ -19,6 +19,7 @@
 Z = 1.96
 
 
+@enum.unique
 class AxPlotTypes(enum.Enum):
     """Enum of Ax plot types."""
 
@@ -28,6 +29,7 @@ class AxPlotTypes(enum.Enum):
     INTERACT_CONTOUR = 3
     BANDIT_ROLLOUT = 4
     INTERACT_SLICE = 5
+    HTML = 6
 
 
 # Configuration for all plots

ax/utils/notebook/plotting.py

@@ -28,6 +28,9 @@ def render(plot_config: AxPlotConfig, inject_helpers=False) -> None:
     """Render plot config."""
     if plot_config.plot_type == AxPlotTypes.GENERIC:
         iplot(plot_config.data)
+    elif plot_config.plot_type == AxPlotTypes.HTML:
+        assert "text/html" in plot_config.data
+        display(plot_config.data, raw=True)
     else:
         display_bundle = {
             "text/html": plot_config_to_html(plot_config, inject_helpers=inject_helpers)