updating exception tests

capitalone · Jul 31, 2024 · e82c4f2 · e82c4f2
1 parent 7f200b0
commit e82c4f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rubicon_ml/viz/dataframe_plot.py b/rubicon_ml/viz/dataframe_plot.py
@@ -125,7 +125,7 @@ def load_experiment_data(self):
                 len(self.experiments),
             )
         if self.data_df is None:
-            raise RubiconException(f"No dataframe with name {self.dataframe_name} found!")
+            raise Exception(f"No dataframe with name {self.dataframe_name} found!")
 
     def register_callbacks(self, link_experiment_table=False):
         outputs = [