fixing tests

automl · Jul 18, 2024 · 8b211a6 · 8b211a6
1 parent ef58978
commit 8b211a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tabpfn_client/tests/unit/test_client.py b/tabpfn_client/tests/unit/test_client.py
@@ -237,8 +237,8 @@ def test_input_data_check(self):
 
         with self.assertRaises(AssertionError) as cm:
             ServiceClient.check_training_data(X[:10000], y[:10000])
-        self.assertEqual(str(cm.exception), "The number of samples should not be more than 10000.")
+        self.assertEqual(str(cm.exception), "The number of features should not be more than 500.")
 
         with self.assertRaises(AssertionError) as cm:
             ServiceClient.check_training_data(X[:, :500], y)
-        self.assertEqual(str(cm.exception), "The number of features should not be more than 500.")
+        self.assertEqual(str(cm.exception), "The number of samples should not be more than 10000.")