update

guestrin-lab · Dec 27, 2024 · 53ec040 · 53ec040
1 parent e403921
commit 53ec040
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/tests/lm_tests.py b/.github/tests/lm_tests.py
@@ -518,8 +518,8 @@ def test_operator_cache(setup_models, model):
     second_response = df.sem_map(user_instruction)
     assert lm.stats.total_usage.operator_cache_hits == 1
 
-    result_1 = first_response["_map"].str.replace(r"[^a-zA-Z\s]", "", regex=True)
-    result_2 = second_response["_map"].str.replace(r"[^a-zA-Z\s]", "", regex=True)
+    result_1 = first_response[first_response["_map"].str.replace(r"[^a-zA-Z\s]", "", regex=True)]
+    result_2 = second_response[second_response["_map"].str.replace(r"[^a-zA-Z\s]", "", regex=True)]
 
     pd.testing.assert_frame_equal(result_1, result_2)
     pd.testing.assert_frame_equal(result_1, expected_response)