Skip to content

Commit 299a650

Browse files
author
AzeezIsh
committed
fixed constants imports
1 parent f4543c0 commit 299a650

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

arrayfire/array_api/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44

55
__all__ = ["__array_api_version__"]
66

7-
from ._constants import Device
7+
from ._constants import Device, e, inf, nan, pi, newaxis
88

9-
__all__ += ["Device"]
9+
__all__ += ["Device", "e", "inf", "nan", "pi", "newaxis"]
1010

1111
from ._creation_function import (
1212
arange,

arrayfire/array_api/_constants.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,3 +105,11 @@ def __post_init__(self) -> None:
105105

106106
class SupportsDLPack(Protocol):
107107
def __dlpack__(self, /, *, stream: None = ...) -> PyCapsule: ...
108+
109+
import math
110+
111+
e = math.e
112+
inf = math.inf
113+
nan = math.nan
114+
pi = math.pi
115+
newaxis = None

0 commit comments

Comments
 (0)