diff --git a/azureml/conda.yml b/azureml/conda.yml index fa4aa98..c69a214 100644 --- a/azureml/conda.yml +++ b/azureml/conda.yml @@ -17,3 +17,4 @@ dependencies: - nltk # This works, while installing from pytorch and cuda from conda does not - torch==2.0.1 + - sentence_transformers>=2.3.1 diff --git a/tests/test_metrics.py b/tests/test_metrics.py index 8983862..73782c0 100644 --- a/tests/test_metrics.py +++ b/tests/test_metrics.py @@ -63,7 +63,7 @@ def test_semscore() -> None: assert semscore >= 0.6, f"SemScore is {semscore}" semscore = eval_semscore(labels, labels) - assert semscore == 1, f"SemScore is {semscore}" + assert semscore == pytest.approx(1.0), f"SemScore is {semscore}" semscore = eval_semscore([], []) assert semscore == 0, f"SemScore is {semscore}"