* replace immutables.Map with immutabledict

* Use Mapping rather than any concrete type in public type annotations

---------

Co-authored-by: Andreas Kloeckner <inform@tiker.net>

* Deprecate reverse_graph in favor of pytools version

* remove it and clean up tests

* merge cleanup

* more comprehensive tests

* minor UsersCollector fixes, remove nonfunctional test part

UsersCollector and NUsersCollector have slightly different
behavior regarding duplicates (ignore in UsersCollector but
counted in NUsersCollector), hence the previous test failed.

* fix flake8

* Update __init__.py

* restore reverse_graph in test

* add progress indicator comment

Co-authored-by: Andreas Kloeckner <inform@tiker.net>

* Rename EinsumAxisDescriptor subclasses.

In order to avoid coming off as something that is more generally applicable to
any pt.Array node.

* adds helpers to get all reduction induction variables

* implements ReductionDescriptor

* defines IndexLambda.var_to_reduction_descr

* defines Einsum.redn_axis_to_redn_descr

* test reduction_axis

* Propagates loopy iname tags for ReductionAxis

* tests loopy tag propagation for ReductionAxis

* introduces ReductionAxis/EinsumIndex specific exception types

* prefer immutables.Map over pyrsistent.PMap

Kaushik Kulkarni authored 3 years ago and Andreas Klöckner committed 3 years ago

- adds py.typed
- adds py.typed path to package_data