fix tox issues

wandb · Sep 22, 2020 · bb41563 · bb41563
1 parent e6296b9
commit bb41563
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/wandb/sdk/wandb_run.py b/wandb/sdk/wandb_run.py
@@ -33,7 +33,7 @@
 from wandb.interface.summary_record import SummaryRecord
 from wandb.lib import config_util, filenames, module, proto_util, redirect, sparkline
 from wandb.util import add_import_hook, sentry_set_scope, to_forward_slash_path
-from wandb.viz import Visualize, CustomChart, custom_plot_on_table
+from wandb.viz import custom_plot_on_table, CustomChart, Visualize
 
 from . import wandb_config
 from . import wandb_history

diff --git a/wandb/sdk_py27/wandb_run.py b/wandb/sdk_py27/wandb_run.py
@@ -33,7 +33,7 @@
 from wandb.interface.summary_record import SummaryRecord
 from wandb.lib import config_util, filenames, module, proto_util, redirect, sparkline
 from wandb.util import add_import_hook, sentry_set_scope, to_forward_slash_path
-from wandb.viz import Visualize, CustomChart, custom_plot_on_table
+from wandb.viz import custom_plot_on_table, CustomChart, Visualize
 
 from . import wandb_config
 from . import wandb_history

diff --git a/wandb/viz.py b/wandb/viz.py
@@ -16,7 +16,7 @@ class CustomChart:
     def __init__(self, panel_config):
         self.panel_config = panel_config
 
-def custom_plot_on_table(vega_spec_name: str, table_key: str, config_mapping: dict):
+def custom_plot_on_table(vega_spec_name, table_key, config_mapping):
     userQuery = {
         "userQuery": {
             "queryFields": [