Skip to content

Commit 3a65664

Browse files
committed
Correct spacing bug
1 parent e7f17d5 commit 3a65664

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

neo4j_parallel_spark_loader/monopartite/batching.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def create_ingest_batches_from_groups(spark_dataframe: DataFrame) -> DataFrame:
3232

3333
coloring = color_complete_graph_with_self_loops(group_count)
3434

35-
coloring_data = [(f"{k[0]}--{k[1]}", v) for k, v in coloring.items()]
35+
coloring_data = [(f"{k[0]} -- {k[1]}", v) for k, v in coloring.items()]
3636

3737
# Create a DataFrame from the coloring dictionary
3838
coloring_df = spark.createDataFrame(coloring_data, ["group", "batch"])

tests/unit/monopartite/conftest.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66
@pytest.fixture(scope="module")
77
def monopartite_batching_data() -> List[Dict[str, int]]:
88
return [
9-
{"group": "1--3", "source_group": 1, "target_group": 3},
10-
{"group": "2--4", "source_group": 2, "target_group": 4},
11-
{"group": "3--5", "source_group": 5, "target_group": 3},
12-
{"group": "4--6", "source_group": 6, "target_group": 4},
13-
{"group": "0--0", "source_group": 0, "target_group": 0},
9+
{"group": "1 -- 3", "source_group": 1, "target_group": 3},
10+
{"group": "2 -- 4", "source_group": 2, "target_group": 4},
11+
{"group": "3 -- 5", "source_group": 5, "target_group": 3},
12+
{"group": "4 -- 6", "source_group": 6, "target_group": 4},
13+
{"group": "0 -- 0", "source_group": 0, "target_group": 0},
1414
]
1515

1616

@@ -20,56 +20,56 @@ def monopartite_dupe_batching_data() -> List[Dict[str, int]]:
2020
{
2121
"source_group": 1,
2222
"target_group": 3,
23-
"group": "1--3",
23+
"group": "1 -- 3",
2424
"source_node": 1,
2525
"target_node": 3,
2626
},
2727
{
2828
"source_group": 2,
2929
"target_group": 4,
30-
"group": "2--4",
30+
"group": "2 -- 4",
3131
"source_node": 4,
3232
"target_node": 2,
3333
},
3434
{
3535
"source_group": 3,
3636
"target_group": 5,
37-
"group": "3--5",
37+
"group": "3 -- 5",
3838
"source_node": 3,
3939
"target_node": 5,
4040
},
4141
{
4242
"source_group": 4,
4343
"target_group": 6,
44-
"group": "4--6",
44+
"group": "4 -- 6",
4545
"source_node": 4,
4646
"target_node": 6,
4747
},
4848
{
4949
"source_group": 0,
5050
"target_group": 0,
51-
"group": "0--0",
51+
"group": "0 -- 0",
5252
"source_node": 0,
5353
"target_node": 0,
5454
},
5555
{
5656
"source_group": 3,
5757
"target_group": 1,
58-
"group": "1--3",
58+
"group": "1 -- 3",
5959
"source_node": 3,
6060
"target_node": 1,
6161
},
6262
{
6363
"source_group": 0,
6464
"target_group": 0,
65-
"group": "0--0",
65+
"group": "0 -- 0",
6666
"source_node": 0,
6767
"target_node": 0,
6868
},
6969
{
7070
"source_group": 6,
7171
"target_group": 4,
72-
"group": "4--6",
72+
"group": "4 -- 6",
7373
"source_node": 6,
7474
"target_node": 4,
7575
},

0 commit comments

Comments
 (0)