From 6e704ffc2d036cf634c08d3c3f30dde51adcdcd5 Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Tue, 11 Feb 2025 14:00:10 -0500 Subject: [PATCH] =?UTF-8?q?=F0=9F=92=A5=20boom:=20make=20private=20files?= =?UTF-8?q?=20private?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Nathaniel Starkman --- src/diffraxtra/__init__.py | 7 +++++-- src/diffraxtra/_src/__init__.py | 13 +++++++++++++ src/diffraxtra/{ => _src}/diffeq.py | 0 src/diffraxtra/{ => _src}/interp.py | 0 4 files changed, 18 insertions(+), 2 deletions(-) create mode 100644 src/diffraxtra/_src/__init__.py rename src/diffraxtra/{ => _src}/diffeq.py (100%) rename src/diffraxtra/{ => _src}/interp.py (100%) diff --git a/src/diffraxtra/__init__.py b/src/diffraxtra/__init__.py index 1644568..9391555 100644 --- a/src/diffraxtra/__init__.py +++ b/src/diffraxtra/__init__.py @@ -6,9 +6,12 @@ "VectorizedDenseInterpolation", ] +from ._src import ( + AbstractVectorizedDenseInterpolation, + DiffEqSolver, + VectorizedDenseInterpolation, +) from ._version import ( # noqa: F401 version as __version__, version_tuple as __version_tuple__, ) -from .diffeq import DiffEqSolver -from .interp import AbstractVectorizedDenseInterpolation, VectorizedDenseInterpolation diff --git a/src/diffraxtra/_src/__init__.py b/src/diffraxtra/_src/__init__.py new file mode 100644 index 0000000..904036f --- /dev/null +++ b/src/diffraxtra/_src/__init__.py @@ -0,0 +1,13 @@ +"""Extras for `diffrax`. Private API.""" + +__all__ = [ + "DiffEqSolver", + "AbstractVectorizedDenseInterpolation", + "VectorizedDenseInterpolation", +] + +from .diffeq import DiffEqSolver +from .interp import ( + AbstractVectorizedDenseInterpolation, + VectorizedDenseInterpolation, +) diff --git a/src/diffraxtra/diffeq.py b/src/diffraxtra/_src/diffeq.py similarity index 100% rename from src/diffraxtra/diffeq.py rename to src/diffraxtra/_src/diffeq.py diff --git a/src/diffraxtra/interp.py b/src/diffraxtra/_src/interp.py similarity index 100% rename from src/diffraxtra/interp.py rename to src/diffraxtra/_src/interp.py