Skip to content

Commit

Permalink
fuel_transactions pep updates
Browse files Browse the repository at this point in the history
  • Loading branch information
nsryan2 committed Jul 26, 2024
1 parent 9874680 commit fb36ffa
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions scripts/fuel_transactions.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from cymetric import timeseries


def used_fuel_transactions(transactions, fuels):
"""
Adds up all the used fuel transactions for each fuel type in a new column.
Expand Down Expand Up @@ -104,7 +105,9 @@ def fuel_received(evaler, fuels, receivers):
if len(fuels) > 1:
for fuel in range(1, len(fuels)):
received_next = timeseries.transactions(
evaler=evaler, receivers=[f'{receiver}'], commodities=[fuels[fuel]])
evaler=evaler,
receivers=[f'{receiver}'],
commodities=[fuels[fuel]])

received[f'{fuels[fuel]}_{receiver}'] = received_next['Mass']

Expand Down Expand Up @@ -151,7 +154,9 @@ def fuel_sent(evaler, fuels, senders):
if len(fuels) > 1:
for fuel in range(1, len(fuels)):
sent_next = timeseries.transactions(
evaler=evaler, senders=[f'{sender}'], commodities=[fuels[fuel]])
evaler=evaler,
senders=[f'{sender}'],
commodities=[fuels[fuel]])

sent[f'{fuels[fuel]}_{sender}'] = sent_next['Mass']

Expand All @@ -160,4 +165,4 @@ def fuel_sent(evaler, fuels, senders):
sent[f'{fuel}_{sender}_total'] = \
sent[f'{fuel}_{sender}'].cumsum()

return sent
return sent

0 comments on commit fb36ffa

Please sign in to comment.