Reorganize ops (#363)

* Split ops into ops.op and ops.builtin * Split ops.array out of ops.builtin * Replace Dispatcher with Op * Clean up docs
pyro-ppl · Sep 14, 2020 · 0d3553b · 0d3553b
1 parent 41735ec
commit 0d3553b
Show file tree

Hide file tree

Showing 9 changed files with 676 additions and 631 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -194,7 +194,8 @@
     'numpy': ('http://docs.scipy.org/doc/numpy/', None),
     'torch': ('http://pytorch.org/docs/master/', None),
     'pyro': ('http://docs.pyro.ai/en/stable/', None),
-    'opt_einsum': ('https://optimized-einsum.readthedocs.io/en/stable/', None)
+    'opt_einsum': ('https://optimized-einsum.readthedocs.io/en/stable/', None),
+    'multipledispatch': ('https://multiple-dispatch.readthedocs.io/en/latest/', None),
 }
 
 # @jpchen's hack to get rtd builder to install latest pytorch

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -11,8 +11,8 @@ Funsor is a tensor-like library for functions and distributions
    :maxdepth: 2
    :caption: Funsor Core:
 
-   domains
    ops
+   domains
    interpretations
    funsors
    optimizer

diff --git a/docs/source/ops.rst b/docs/source/ops.rst
@@ -1,6 +1,25 @@
 Operations
-----------
-.. automodule:: funsor.ops
+==========
+
+Operation classes
+-----------------
+.. automodule:: funsor.ops.op
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :member-order: bysource
+
+Builtin operations
+------------------
+.. automodule:: funsor.ops.builtin
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :member-order: bysource
+
+Array operations
+----------------
+.. automodule:: funsor.ops.array
     :members:
     :undoc-members:
     :show-inheritance: