PR comments

Author: nicolaskruchten

packages/python/plotly/plotly/data/__init__.py

@@ -117,11 +117,11 @@ def timeseries():
 def experiment():
     """
 Each row in this wide dataset represents the results of 100 simulated participants
-on three hypothetical experiments, along with their gender and smoker status.
+on three hypothetical experiments, along with their gender and control/treatment group.
 
 Returns:
     A `pandas.DataFrame` with 100 rows and the following columns:
-    `['experiment_1', 'experiment_2', 'experiment_3', 'gender', 'smoker']`.
+    `['experiment_1', 'experiment_2', 'experiment_3', 'gender', 'group']`.
 """
     return _get_dataset("experiment")
 

packages/python/plotly/plotly/express/_core.py

@@ -936,6 +936,10 @@ def _get_reserved_col_names(args):
 
 
 def _is_col_list(df_input, arg):
+    """Returns True if arg looks like it's a list of columns or references to columns
+    in df_input, and False otherwise (in which case it's assumed to be a single column
+    or reference to a column).
+    """
     if arg is None or isinstance(arg, str) or isinstance(arg, int):
         return False
     if isinstance(arg, pd.MultiIndex):