Skip to content

Commit ff23e72

Browse files
authored
Merge pull request #113 from OpenBioSim/fix_perturbed_system
Allow perturbed system option to be a Sire system and a string
2 parents 612edaf + 46faa86 commit ff23e72

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/somd2/config/_config.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1624,7 +1624,10 @@ def perturbed_system(self):
16241624
@perturbed_system.setter
16251625
def perturbed_system(self, perturbed_system):
16261626
if perturbed_system is not None:
1627-
if isinstance(perturbed_system, str):
1627+
if isinstance(perturbed_system, _sr.system.System):
1628+
self._perturbed_system = perturbed_system
1629+
self._perturbed_system_file = None
1630+
elif isinstance(perturbed_system, str):
16281631
import os
16291632

16301633
if not os.path.exists(perturbed_system):
@@ -1640,7 +1643,9 @@ def perturbed_system(self, perturbed_system):
16401643
f"Unable to load 'perturbed_system' stream file: {e}"
16411644
)
16421645
else:
1643-
raise TypeError("'perturbed_system' must be of type 'str'")
1646+
raise TypeError(
1647+
"'perturbed_system' must be of type 'sr.system.System' or 'str'"
1648+
)
16441649
else:
16451650
self._perturbed_system = None
16461651
self._perturbed_system_file = None

0 commit comments

Comments
 (0)