Use list of columns

Raul-Ciria-Aylagas · Apr 9, 2019 · cb7b910 · cb7b910
1 parent ee31e63
commit cb7b910
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/trnpy/misc.py b/trnpy/misc.py
@@ -211,7 +211,7 @@ def bokeh_circles_from_df(df_in, x_col, y_cols=[], tips_cols=[], size=10,
     ``plot_width``, etc.
     '''
     if len(y_cols) == 0:  # Per default, use all columns in the DataFrame
-        y_cols = df_in.columns
+        y_cols = list(df_in.columns)
 
     df = df_in.reset_index()  # Remove index
     selection = y_cols + [x_col] + list(tips_cols)
@@ -321,7 +321,6 @@ def bokeh_time_line(df_in, y_cols=[], palette=palette_default,
     '''
     if len(y_cols) == 0:  # Per default, use all columns in the DataFrame
         y_cols = list(df_in.columns)
-    x_col = x_col
 
     df = df_in.reset_index()  # Remove index
     source = ColumnDataSource(data=df[[x_col]+y_cols])  # Use required columns

diff --git a/polyfit.py → trnpy/polyfit.py b/polyfit.py → trnpy/polyfit.py