Skip to content

Commit f22edc0

Browse files
Drop Aesara support
This simplifies testing & upcoming fixes related to changes in shape handling in PyTensor.
1 parent 5050779 commit f22edc0

File tree

3 files changed

+7
-15
lines changed

3 files changed

+7
-15
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def get_version():
2020
author='Adrian Seyboldt',
2121
author_email='[email protected]',
2222
description='Python wrapper of sundials for solving ordinary differential equations',
23-
url='https://github.com/aseyboldt/sunode',
23+
url='https://github.com/pymc-devs/sunode',
2424
setup_requires=["cffi>=1.0.0"],
2525
cffi_modules=[
2626
"sunode/build_cvodes.py:ffibuilder",

sunode/wrappers/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
11
from . import as_pytensor
2-
from . import as_pytensor as as_aesara
32

4-
__all__ = ['as_aesara', 'as_pytensor']
3+
__all__ = ("as_pytensor",)

sunode/wrappers/as_pytensor.py

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,8 @@
1-
try:
2-
import pytensor.tensor as pt
3-
from pytensor.graph.basic import Constant, Variable
4-
from pytensor.graph.fg import MissingInputError
5-
from pytensor.graph.op import Op
6-
from pytensor.gradient import grad_not_implemented
7-
except ModuleNotFoundError:
8-
import aesara.tensor as pt
9-
from aesara.graph.basic import Constant, Variable
10-
from aesara.graph.fg import MissingInputError
11-
from aesara.graph.op import Op
12-
from aesara.gradient import grad_not_implemented
1+
import pytensor.tensor as pt
2+
from pytensor.graph.basic import Constant, Variable
3+
from pytensor.graph.fg import MissingInputError
4+
from pytensor.graph.op import Op
5+
from pytensor.gradient import grad_not_implemented
136
import copy
147
from typing import Dict, Optional, Any, Callable
158

0 commit comments

Comments
 (0)