Skip to content

Commit 0c1168b

Browse files
committed
Removed ensemble description from __str__ to make print calls on ThermoParameters behave more intuitively
1 parent 6354a12 commit 0c1168b

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

polymerist/mdtools/openmmtools/thermo.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def ensemble(self) -> Ensemble:
153153
return Ensemble.NPT
154154
return Ensemble.NVT
155155

156-
def __str__(self) -> str:
156+
def describe_ensemble(self) -> str:
157157
'''Verbal description of ensemble'''
158158
return f'{self.ensemble.name} ({self.ensemble.value.capitalize()}) ensemble'
159159

@@ -164,7 +164,7 @@ def integrator(self, time_step : Quantity) -> Integrator:
164164
else:
165165
integrator = VerletIntegrator(time_step)
166166

167-
LOGGER.info(f'Created {integrator.__class__.__name__} for {self!s}')
167+
LOGGER.info(f'Created {integrator.__class__.__name__} for {self.describe_ensemble()}')
168168

169169
return integrator
170170

@@ -177,6 +177,6 @@ def forces(self) -> Optional[Iterable[Force]]:
177177
forces.extend(self.barostat_params.forces())
178178

179179
for force in forces:
180-
LOGGER.info(f'Created {force.__class__.__name__} Force for {self!s}')
180+
LOGGER.info(f'Created {force.__class__.__name__} Force for {self.describe_ensemble()}')
181181

182182
return forces

0 commit comments

Comments
 (0)