Skip to content

Commit 577520b

Browse files
committed
Avoid usage of dummy password
1 parent d97a138 commit 577520b

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

graphdatascience/graph_data_science.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,6 @@ def __init__(
9191
with open(self._path("graphdatascience.resources.field-testing", "pub.pem"), "rb") as f:
9292
pub_key = rsa.PublicKey.load_pkcs1(f.read())
9393
self._encrypted_db_password = rsa.encrypt(auth[1].encode(), pub_key).hex()
94-
else:
95-
self._encrypted_db_password = '12345'
9694

9795
self._compute_cluster_ip = None
9896

@@ -156,7 +154,7 @@ def kge(self) -> KgeRunner:
156154
"gds.kge.model",
157155
self._server_version,
158156
self._compute_cluster_ip,
159-
self._encrypted_db_password,
157+
self._encrypted_db_password if self._query_runner.encrypted() else None,
160158
self._query_runner._gds_arrow_client._host + ":" + str(self._query_runner._gds_arrow_client._port),
161159
)
162160

graphdatascience/model/kge_runner.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,13 @@ def train(
6666
"task": "KGE_TRAINING_PYG",
6767
"task_config": {
6868
"graph_config": graph_config,
69-
"modelname": "dummmy_model_name",
69+
"modelname": "dummmy_model_name"+str(time.time()),
7070
"task_config": algo_config,
7171
},
72-
# "encrypted_db_password": self._encrypted_db_password,
7372
"graph_arrow_uri": self._arrow_uri,
7473
}
74+
if self._encrypted_db_password is not None:
75+
config["encrypted_db_password"] = self._encrypted_db_password
7576

7677
if mlflow_experiment_name is not None:
7778
config["task_config"]["mlflow"] = {

0 commit comments

Comments
 (0)