remove all dataframes from experiments test

capitalone · Jul 31, 2024 · 146711a · 146711a
1 parent e82c4f2
commit 146711a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/viz/test_dataframe_plot.py b/tests/unit/viz/test_dataframe_plot.py
@@ -79,8 +79,8 @@ def test_dataframe_plot_register_callbacks_link(viz_experiments, is_linked, expe
 
 def test_dataframe_not_logged_all_experiments(viz_experiments):
     for exp in viz_experiments:
-        if len(exp.dataframes()) == 0:
+        if len(exp.dataframes()) != 0:
             viz_experiments.remove(exp)
     dataframe_plot = DataframePlot("test dataframe", experiments=viz_experiments)
-    with pytest.raises(RubiconException):
+    with pytest.raises(Exception):
         assert dataframe_plot.load_experiment_data()