refactor!: implement Model interface to lambdify Callables

.gitignore

@@ -3,6 +3,7 @@
 *.json
 *.npy
 *.pdf
+*.pickle
 *.png
 *.svg
 *.v2

.pre-commit-config.yaml

@@ -37,6 +37,8 @@ repos:
     hooks:
       - id: nbqa-black
         args: [--nbqa-mutate]
+      - id: nbqa-flake8
+        args: ["--extend-ignore=E402,F821"]
       - id: nbqa-isort
         args: [--nbqa-mutate]
 

.vscode/settings.json

@@ -69,10 +69,10 @@
   "telemetry.enableCrashReporter": false,
   "telemetry.enableTelemetry": false,
   "yaml.schemas": {
-    "https://raw.githubusercontent.com/ComPWA/expertsystem/0.6.10/src/expertsystem/schemas/amplitude-model.json": [
+    "https://raw.githubusercontent.com/ComPWA/expertsystem/0.7.0a0/src/expertsystem/schemas/amplitude-model.json": [
       "*recipe.y*ml"
     ],
-    "https://raw.githubusercontent.com/ComPWA/expertsystem/0.6.10/src/expertsystem/schemas/particle-list.json": [
+    "https://raw.githubusercontent.com/ComPWA/expertsystem/0.7.0a0/src/expertsystem/schemas/particle-list.json": [
       "*particle.y*ml"
     ]
   }

README.md

@@ -18,18 +18,13 @@
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort)
 
-TensorWaves is a Python package for doing Partial Wave Analysis with
-[TensorFlow](https://www.tensorflow.org/) as computational backend. The package
-originates from [pycompwa](https://compwa.github.io/), which did its
-computations through ([ComPWA](https://github.com/ComPWA/ComPWA)) ― ComPWA's
-function tree is similar to TensorFlow graphs and can therefore be replaced by
-the tools that TensorFlow already offers. The decision to completely migrate
-ComPWA to TensorFlow has been further inspired by
-[TensorFlowAnalysis](https://gitlab.cern.ch/poluekt/TensorFlowAnalysis) and
-[zfit](https://github.com/zfit/zfit/).
+TensorWaves is a fitter package that optimizes mathematical models to data
+samples. The models can be any mathematical expression that is then converted
+to any computational backend. In addition, TensorWaves provides functionality
+to generate toy Monte Carlo data samples.
 
 All documentation can be found on
-[tensorwaves.rtfd.io](https://pwa.readthedocs.io/projects/tensorwaves/).
+[tensorwaves.rtfd.io](https://pwa.readthedocs.io/projects/tensorwaves).
 
 For an overview of **upcoming releases and planned functionality**, see
 [here](https://github.com/ComPWA/tensorwaves/milestones?direction=asc&sort=title&state=open).

cspell.json

@@ -47,6 +47,7 @@
     ],
     "language": "en-US",
     "words": [
+        "backends",
         "blatt",
         "bottomness",
         "breit",
@@ -72,6 +73,8 @@
         "itertools",
         "jsonschema",
         "jupyter",
+        "lambdifying",
+        "lineshape",
         "mathbb",
         "matplotlib",
         "mkdir",
@@ -97,17 +100,20 @@
         "topness",
         "traceback",
         "unbinned",
+        "vectorize",
         "venv",
         "weisskopf",
         "xcode",
         "zfit"
     ],
     "ignoreWords": [
         "amplitf",
+        "arange",
         "atfi",
         "axvline",
         "cano",
         "celltoolbar",
+        "cmin",
         "codacy",
         "codecov",
         "codemirror",
@@ -118,18 +124,24 @@
         "dtype",
         "eval",
         "evalf",
+        "figsize",
         "fmin",
         "fval",
         "genindex",
+        "gridspec",
         "heli",
         "histtype",
         "iloc",
+        "imag",
         "iminuit",
         "indeterministic",
         "isort",
         "jupyterlab",
         "keras",
         "kernelspec",
+        "lambdification",
+        "lambdified",
+        "lambdify",
         "linestyle",
         "linkcheck",
         "linspace",
@@ -143,11 +155,13 @@
         "mprime",
         "nbconvert",
         "nbformat",
+        "nbins",
         "nbody",
         "nbsphinx",
         "ncalls",
         "ndarray",
         "noqa",
+        "numba",
         "pandoc",
         "phasespace",
         "phsp",
@@ -162,11 +176,18 @@
         "pytestconfig",
         "rightarrow",
         "rtfd",
+        "scipy",
         "seealso",
+        "sharex",
         "subsys",
+        "tolist",
         "tqdm",
+        "unflatten",
+        "unflattened",
         "unnormalized",
         "vstack",
-        "xlabel"
+        "xlabel",
+        "xlim",
+        "ylim"
     ]
 }

docs/conf.py

@@ -135,17 +135,19 @@
 # Intersphinx settings
 intersphinx_mapping = {
     "expertsystem": (
-        "https://pwa.readthedocs.io/projects/expertsystem/en/0.6.10",
+        "https://pwa.readthedocs.io/projects/expertsystem/en/0.7.0a0",
         None,
     ),
     "iminuit": ("https://iminuit.readthedocs.io/en/stable", None),
+    "jax": ("https://jax.readthedocs.io/en/stable", None),
     "matplotlib": ("https://matplotlib.org", None),
     "mypy": ("https://mypy.readthedocs.io/en/stable", None),
     "numpy": ("https://numpy.org/doc/stable", None),
     "pandas": ("https://pandas.pydata.org/pandas-docs/stable", None),
     "pwa": ("https://pwa.readthedocs.io", None),
     "pycompwa": ("https://compwa.github.io", None),
     "python": ("https://docs.python.org/3", None),
+    "sympy": ("https://docs.sympy.org/latest", None),
     "tensorflow": (
         "https://www.tensorflow.org/api_docs/python",
         "tensorflow.inv",

docs/index.md

@@ -16,17 +16,27 @@ For an overview of upcoming releases and planned functionality, see
 ```
 ````
 
-For a quick demo of the TensorWaves workflow, see {doc}`/usage`.
-
-TensorWaves is a Python package for doing Partial Wave Analysis with
-[TensorFlow](https://www.tensorflow.org) as computational backend. The package
-originates from [pycompwa](pycompwa), which did its computations through
-[ComPWA](https://github.com/ComPWA/ComPWA) — ComPWA's function tree is similar
-to TensorFlow graphs and can therefore be replaced by the tools that TensorFlow
-already offers. The decision to completely migrate ComPWA to TensorFlow has
-been further inspired by
+TensorWaves is a fitter package that optimizes mathematical models to data
+samples. The models can be any mathematical expression (best expressed with
+{mod}`sympy`) that is then converted to any computational backend. In addition,
+the {mod}`tensorwaves.data` module allows one to generate toy data samples.
+
+```{link-button} usage
+---
+classes: btn-outline-primary btn-block
+type: ref
+text: Click here for a quick demo
+---
+```
+
+The package originates from {mod}`pycompwa`, which did its computations through
+[ComPWA](https://github.com/ComPWA/ComPWA). ComPWA's function tree is similar
+to for instance
+[TensorFlow graphs](https://www.tensorflow.org/tensorboard/graphs) and
+[SymPy expression trees](https://docs.sympy.org/latest/tutorial/manipulation.html).
+The decision to use Python only has been further inspired by
 [TensorFlowAnalysis](https://gitlab.cern.ch/poluekt/TensorFlowAnalysis)/[AmpliTF](https://github.com/apoluekt/AmpliTF)
-and [zfit](https://github.com/zfit/zfit)
+and [zfit](https://github.com/zfit/zfit).
 
 ## Table of Contents
 

docs/tensorflow.txt

@@ -3,6 +3,7 @@
 # Version: 2
 # The remainder of this file is compressed using zlib.
 
+tf.Tensor   py:class 1 tf/Tensor -
 tf.Variable py:class 1 tf/Variable -
 tensorflow.python.keras.losses.Loss py:class 1 tf/keras/losses/Loss -
 tensorflow.python.ops.stateful_random_ops.Generator py:class 1 tf/random/Generator -