Add power as argument to TransportOperator.__call__

This commit is contained in:
Andrew Johnson 2019-08-06 16:57:21 -05:00
parent 49ba67d480
commit 78943d1086
No known key found for this signature in database
GPG key ID: 253418E91B7F6FEB

View file

@ -106,13 +106,15 @@ class TransportOperator(ABC):
self._dilute_initial = value
@abstractmethod
def __call__(self, vec):
def __call__(self, vec, power):
"""Runs a simulation.
Parameters
----------
vec : list of numpy.ndarray
Total atoms to be used in function.
power : float
Power of the reactor in [W]
Returns
-------