Skip to content

Commit 4d329ac

Browse files
committed
MAINT: Make linting happy and add numba to poetry
1 parent e99ed93 commit 4d329ac

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

benchmarks/benchmarks.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from decimal import Decimal
22

33
import numpy as np
4+
45
import numpy_financial as npf
56

67

@@ -31,15 +32,15 @@ def time_broadcast(self, n_cashflows, cashflow_lengths, rates_lengths):
3132
npf.npv(self.rates, self.cashflows)
3233

3334
def time_for_loop(self, n_cashflows, cashflow_lengths, rates_lengths):
34-
for i, rate in enumerate(self.rates):
35-
for j, cashflow in enumerate(self.cashflows):
35+
for rate in self.rates:
36+
for cashflow in self.cashflows:
3637
npf.npv(rate, cashflow)
3738

3839
def time_broadcast_decimal(self, n_cashflows, cashflow_lengths, rates_lengths):
3940
npf.npv(self.rates_decimal, self.cashflows_decimal)
4041

4142
def time_for_loop_decimal(self, n_cashflows, cashflow_lengths, rates_lengths):
42-
for i, rate in enumerate(self.rates_decimal):
43-
for j, cashflow in enumerate(self.cashflows_decimal):
43+
for rate in self.rates_decimal:
44+
for cashflow in self.cashflows_decimal:
4445
npf.npv(rate, cashflow)
4546

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ packages = [{include = "numpy_financial"}]
3939
[tool.poetry.dependencies]
4040
python = "^3.9,<3.12"
4141
numpy = "^1.23"
42+
numba = "^0.58.1"
4243

4344

4445
[tool.poetry.group.test.dependencies]

0 commit comments

Comments
 (0)