Skip to content

Commit 88452ef

Browse files
authored
Add PySBModel.to_str (#451)
Convenient for debugging.
1 parent 97bf770 commit 88452ef

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

petab/v1/models/pysb_model.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,14 +93,19 @@ def from_file(
9393
)
9494

9595
def to_file(self, filename: str | Path | None = None) -> None:
96-
from pysb.export import export
96+
model_source = self.to_str()
9797

98-
model_source = export(self.model, "pysb_flat")
9998
with open(
10099
filename or _generate_path(self.rel_path, self.base_path), "w"
101100
) as f:
102101
f.write(model_source)
103102

103+
def to_str(self) -> str:
104+
"""Get the PySB model Python code as a string."""
105+
from pysb.export import export
106+
107+
return export(self.model, "pysb_flat")
108+
104109
@property
105110
def model_id(self):
106111
return self._model_id

0 commit comments

Comments
 (0)