Skip to content

Commit

Permalink
changed rm tests index name to avoid pinecone failures
Browse files Browse the repository at this point in the history
  • Loading branch information
AmoghTantradi committed Jan 28, 2025
1 parent ddfd549 commit 20206e1
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions .github/tests/rm_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -189,7 +189,7 @@ def test_vs_cluster_by(setup_models, setup_vs, vs, model):
]
}
df = pd.DataFrame(data)
df = df.sem_index("Course Name", "index_dir")
df = df.sem_index("Course Name", "index-dir")
df = df.sem_cluster_by("Course Name", 2)
groups = df.groupby("cluster_id")["Course Name"].apply(set).to_dict()
assert len(groups) == 2, groups
Expand Down Expand Up @@ -219,7 +219,7 @@ def test_vs_search_rm_only(setup_models, setup_vs, vs, model):
]
}
df = pd.DataFrame(data)
df = df.sem_index("Course Name", "index_dir")
df = df.sem_index("Course Name", "index-dir")
df = df.sem_search("Course Name", "Optimization", K=1)
assert df["Course Name"].tolist() == ["Optimization Methods in Engineering"]

Expand All @@ -240,7 +240,7 @@ def test_vs_sim_join(setup_models, setup_vs, vs, model):
data2 = {"Skill": ["Math", "History"]}

df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2).sem_index("Skill", "index_dir")
df2 = pd.DataFrame(data2).sem_index("Skill", "index-dir")
joined_df = df1.sem_sim_join(df2, left_on="Course Name", right_on="Skill", K=1)
joined_pairs = set(zip(joined_df["Course Name"], joined_df["Skill"]))
expected_pairs = {("History of the Atlantic World", "History"), ("Riemannian Geometry", "Math")}
Expand All @@ -266,7 +266,7 @@ def test_vs_dedup(setup_models, setup_vs, vs):
]
}
df = pd.DataFrame(data)
df = df.sem_index("Text", "index_dir").sem_dedup("Text", threshold=0.85)
df = df.sem_index("Text", "index-dir").sem_dedup("Text", threshold=0.85)
kept = df["Text"].tolist()
kept.sort()
assert len(kept) == 2, kept
Expand Down

0 comments on commit 20206e1

Please sign in to comment.