Skip to content

Commit d623a9a

Browse files
committed
Slightly cleaner way of copying harness db config
1 parent a525ed5 commit d623a9a

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

qcarchivetesting/qcarchivetesting/testing_classes.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from __future__ import annotations
22

33
import logging
4-
from copy import deepcopy
54

65
from qcarchivetesting import geoip_path, geoip_filename, ip_tests_enabled
76
from qcfractal.config import DatabaseConfig
@@ -78,10 +77,8 @@ def __init__(self, db_path: str):
7877
assert self.harness.is_alive() and not self.harness.can_connect()
7978

8079
def get_new_harness(self, db_name: str) -> QCATestingPostgresHarness:
81-
harness_config = deepcopy(self.harness.config.model_dump())
82-
harness_config["database_name"] = db_name
83-
84-
new_harness = QCATestingPostgresHarness(DatabaseConfig(**harness_config))
80+
db_config = self.harness.config.model_copy(deep=True, update={"database_name": db_name})
81+
new_harness = QCATestingPostgresHarness(db_config)
8582
new_harness.create_database(create_tables=True)
8683
return new_harness
8784

0 commit comments

Comments
 (0)