Skip to content

Commit 0ecc81f

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
fix tests
1 parent 0c00965 commit 0ecc81f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

packages/graphrag/graphrag/config/embeddings.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,5 @@ def create_index_name(
3737
raise KeyError(msg)
3838

3939
if index_prefix:
40-
return f"{index_prefix}-{embedding_name}".replace(".", "-")
41-
return embedding_name.replace(".", "-")
40+
return f"{index_prefix}-{embedding_name}"
41+
return embedding_name

tests/unit/utils/test_embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@ def test_create_index_name_invalid_embedding_throws():
1717

1818
def test_create_index_name_invalid_embedding_does_not_throw():
1919
collection = create_index_name("default", "invalid_name", validate=False)
20-
assert collection == "default_invalid-name"
20+
assert collection == "default-invalid_name"

0 commit comments

Comments
 (0)