diff --git a/src/somd2/config/_config.py b/src/somd2/config/_config.py index faac9ad..52babac 100644 --- a/src/somd2/config/_config.py +++ b/src/somd2/config/_config.py @@ -1624,7 +1624,10 @@ def perturbed_system(self): @perturbed_system.setter def perturbed_system(self, perturbed_system): if perturbed_system is not None: - if isinstance(perturbed_system, str): + if isinstance(perturbed_system, _sr.system.System): + self._perturbed_system = perturbed_system + self._perturbed_system_file = None + elif isinstance(perturbed_system, str): import os if not os.path.exists(perturbed_system): @@ -1640,7 +1643,9 @@ def perturbed_system(self, perturbed_system): f"Unable to load 'perturbed_system' stream file: {e}" ) else: - raise TypeError("'perturbed_system' must be of type 'str'") + raise TypeError( + "'perturbed_system' must be of type 'sr.system.System' or 'str'" + ) else: self._perturbed_system = None self._perturbed_system_file = None