From 53ec0401be3335745fe6e66706069b17475052be Mon Sep 17 00:00:00 2001 From: StanChan03 Date: Thu, 26 Dec 2024 21:00:21 -0800 Subject: [PATCH] update --- .github/tests/lm_tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/tests/lm_tests.py b/.github/tests/lm_tests.py index da3dfab..eb1fc22 100644 --- 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)