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