From ecd4109a865d798a49cc4190fb010678b4bb3a62 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 8 Aug 2019 00:51:34 +0200
Subject: [PATCH] Add module mapping pystencils_autodiff -> pystencils.autodiff
---
src/pystencils_autodiff/__init__.py | 5 +++++
src/pystencils_autodiff/backends/__init__.py | 7 +++++++
2 files changed, 12 insertions(+)
diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py
index 752dd6a..7c6f4b9 100644
--- a/src/pystencils_autodiff/__init__.py
+++ b/src/pystencils_autodiff/__init__.py
@@ -1,3 +1,5 @@
+import sys
+
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,
@@ -15,3 +17,6 @@ __all__ = ['backends',
"tf_constant_from_field", " tf_placeholder_from_field",
"tf_scalar_variable_from_field", " tf_variable_from_field",
"torch_tensor_from_field"]
+
+sys.modules['pystencils.autodiff'] = pystencils_autodiff
+sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends
diff --git a/src/pystencils_autodiff/backends/__init__.py b/src/pystencils_autodiff/backends/__init__.py
index d2ca875..d22da0f 100644
--- a/src/pystencils_autodiff/backends/__init__.py
+++ b/src/pystencils_autodiff/backends/__init__.py
@@ -1,2 +1,9 @@
+"""
+Backends for operators to support automatic Differentation
+
+Currently, we can use pystencils' JIT compilation to register
+a Torch or a Tensorflow operation or we can compile a static
+library to be directly loaded into Torch/Tensorflow.
+"""
AVAILABLE_BACKENDS = ['tensorflow', 'torch', 'tensorflow_cpp', 'torch_native']
--
GitLab