File tree Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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" ] = {
You can’t perform that action at this time.
0 commit comments