Dataframe name

rubicon_ml/viz/dataframe_plot.py

@@ -43,17 +43,22 @@ class DataframePlot(VizBase):
 
     def __init__(
         self,
-        dataframe_name,
+        dataframe_name=None,
         experiments=None,
         plotting_func=px.line,
         plotting_func_kwargs={},
         x=None,
         y=None,
     ):
         super().__init__(dash_title="plot dataframes")
-
-        self.dataframe_name = dataframe_name
         self.experiments = experiments
+        if experiments != None:
+            if len(experiments[0].dataframes()) == 0:
+                raise Exception("No dataframe logged to experiment")
+            else:
+                self.dataframe_name = self.experiments[0].dataframes()[0].name
+        else:
+            self.dataframe_name = dataframe_name
         self.plotting_func = plotting_func
         self.plotting_func_kwargs = plotting_func_kwargs
         self.x = x
@@ -89,8 +94,15 @@ def load_experiment_data(self):
         `dataframe_name` must have the same schema.
         """
         self.data_df = None
+        count = 0
 
         for experiment in self.experiments:
+            if len(experiment.dataframes()) == 0:
+                print(
+                    f"WARNING: Experiment {experiment.name} does not have any dataframes logged to it. "
+                )
+                continue
+            count += 1
             dataframe = experiment.dataframe(name=self.dataframe_name)
 
             data_df = dataframe.get_data()
@@ -116,6 +128,9 @@ def load_experiment_data(self):
                 len(self.experiments),
             )
 
+        if count == 0:
+            raise Exception(f"No dataframe with name {self.dataframe_name} found!")
+
     def register_callbacks(self, link_experiment_table=False):
         outputs = [
             Output("dataframe-plot", "figure"),