Skip to content

Commit 69f9239

Browse files
authored
Change 'print' by 'return' in main (#104)
* change print by return so it can be used in scripts
1 parent f47b9b5 commit 69f9239

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

rmsd/calculate_rmsd.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -1877,7 +1877,7 @@ def parse_arguments(arguments: Optional[Union[str, List[str]]] = None) -> argpar
18771877
return args
18781878

18791879

1880-
def main(args: Optional[List[str]] = None) -> None:
1880+
def main(args: Optional[List[str]] = None):
18811881

18821882
# Parse arguments
18831883
settings = parse_arguments(args)
@@ -2051,15 +2051,16 @@ def main(args: Optional[List[str]] = None) -> None:
20512051

20522052
# done and done
20532053
xyz = set_coordinates(q_all_atoms, q_coord, title=f"{settings.structure_b} - modified")
2054-
print(xyz)
2054+
return xyz
20552055

20562056
else:
20572057

20582058
if not result_rmsd:
20592059
result_rmsd = rmsd_method(p_coord, q_coord)
20602060

2061-
print("{0}".format(result_rmsd))
2061+
return result_rmsd
20622062

20632063

20642064
if __name__ == "__main__":
2065-
main() # pragma: no cover
2065+
result = main() # pragma: no cover
2066+
print(result)

0 commit comments

Comments
 (0)