Initial type hints for function_wrapper (pytorch#4947)

* Initial type hints for function_wrapper * Don't break python 2 * Update TopEnvironment * Add mypy check to travis * Add .mypy_cache to .gitignore
kevinch-nv · Feb 8, 2018 · 0629785 · 0629785
1 parent 696db00
commit 0629785
Show file tree

Hide file tree

Showing 3 changed files with 227 additions and 20 deletions.
diff --git a/.gitignore b/.gitignore
@@ -40,6 +40,7 @@ test/.coverage
 */**/*.dylib*
 test/data/legacy_serialized.pt
 test/data/linear.pt
+.mypy_cache
 
 # IPython notebook checkpoints
 .ipynb_checkpoints

diff --git a/.travis.yml b/.travis.yml
@@ -12,7 +12,12 @@ sudo: false
 matrix:
     fast_finish: true
     include:
-        env: LINT_CHECK
+      - env: LINT_CHECK
         python: "2.7"
         install: pip install flake8
         script: flake8
+      # mypy will complain about various files. Limiting it to only typed files
+      - env: MYPY_TYPE_CHECK
+        python: "3.6"
+        install: pip install mypy mypy-extensions
+        script: mypy --py2 aten/src/ATen/function_wrapper.py