fix tests

explodinggradients · Sep 6, 2024 · a15da58 · a15da58
1 parent 150e81b
commit a15da58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/test_metric.py b/tests/unit/test_metric.py
@@ -13,7 +13,7 @@ def test_get_available_metrics():
 
     assert all(
         [
-            m.required_columns == {"SINGLE_TURN": {"response", "user_input"}}
+            m.required_columns["SINGLE_TURN"] == {"response", "user_input"}
             for m in get_available_metrics(ds)
         ]
     ), "All metrics should have required columns ('user_input', 'response')"