typing-inspect>=0.4.0

[:python_version == "3.6"]
dataclasses

[dev]
pytest
ipython
flake8
numpy==1.18.5
torch==1.8.0
