Skip to content

Commit 4bc9519

Browse files
authored
chore: revert the change of session for llm tests (#1625)
* revert the change of session for llm tests * fix a typo
1 parent c37c844 commit 4bc9519

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/system/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1263,10 +1263,10 @@ def penguins_randomforest_classifier_model_name(
12631263

12641264
@pytest.fixture(scope="session")
12651265
def llm_fine_tune_df_default_index(
1266-
test_session: bigframes.Session,
1266+
session: bigframes.Session,
12671267
) -> bigframes.dataframe.DataFrame:
12681268
training_table_name = "llm_tuning.emotion_classification_train"
1269-
df = test_session.read_gbq(training_table_name).dropna().head(30)
1269+
df = session.read_gbq(training_table_name).dropna().head(30)
12701270
prefix = "Please do sentiment analysis on the following text and only output a number from 0 to 5 where 0 means sadness, 1 means joy, 2 means love, 3 means anger, 4 means fear, and 5 means surprise. Text: "
12711271
df["prompt"] = prefix + df["text"]
12721272
df["label"] = df["label"].astype("string")

0 commit comments

Comments
 (0)