diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py index 7c6f4b9fa2650d3c658af2babb1e1099e7d9632c..fafab5ab22e7ac018d4feeac67ef7dd49f339c4a 100644 --- a/src/pystencils_autodiff/__init__.py +++ b/src/pystencils_autodiff/__init__.py @@ -4,8 +4,8 @@ import pystencils_autodiff.backends # NOQA from pystencils_autodiff._field_to_tensors import ( # NOQA tf_constant_from_field, tf_placeholder_from_field, tf_scalar_variable_from_field, tf_variable_from_field, torch_tensor_from_field) -from pystencils_autodiff.adjoint_field import AdjointField -from pystencils_autodiff.autodiff import ( +from pystencils_autodiff._adjoint_field import AdjointField +from pystencils_autodiff._autodiff import ( AutoDiffAstPair, AutoDiffOp, create_backward_assignments, get_jacobian_of_assignments) __all__ = ['backends', diff --git a/src/pystencils_autodiff/adjoint_field.py b/src/pystencils_autodiff/_adjoint_field.py similarity index 100% rename from src/pystencils_autodiff/adjoint_field.py rename to src/pystencils_autodiff/_adjoint_field.py diff --git a/src/pystencils_autodiff/autodiff.py b/src/pystencils_autodiff/_autodiff.py similarity index 100% rename from src/pystencils_autodiff/autodiff.py rename to src/pystencils_autodiff/_autodiff.py