guess_type fixups (#53)

JelleZijlstra · web-flow · commit 17bd815ee99c · 2023-01-20T05:10:23.000-08:00
- Shorter name for the file
- Add docs
- Remove unused import
diff --git a/README.md b/README.md
@@ -37,6 +37,9 @@ more transformations. The following are supported:
     default that is named `foo` with `bar.Baz`. For example, use
     `--annotate-named-param uid:my_types.Uid` to annotate any `uid`
     parameter in your codebase with no default as `my_types.Uid`.
+  - `--guess-common-names`: infer certain parameter types from their names
+    based on common patterns in open-source Python code. For example, infer
+    that a `verbose` parameter is of type `bool`.
 - Annotating magical methods:
   - `--annotate-magics`: add type annotation to certain magic methods.
     Currently this does the following:
@@ -97,7 +100,6 @@ There are two shortcut flags to enable multiple transformations at once:
 - Do not add `None` return types to methods marked with `@abstractmethod` and
   to methods in stub files
 - Improve type inference:
-
   - `"string" % ...` is always `str`
   - `b"bytes" % ...` is always `bytes`
   - An `and` or `or` operator where left and right sides are of the same type
diff --git a/autotyping/autotyping.py b/autotyping/autotyping.py
@@ -10,7 +10,7 @@
 from libcst.codemod.visitors import AddImportsVisitor
 from libcst.metadata import CodePosition, CodeRange, PositionProvider
 
-from autotyping.guess_type_from_argname import guess_type_from_argname
+from autotyping.guess_type import guess_type_from_argname
 
 _DEFAULT_POSITION = CodePosition(0, 0)
 _DEFAULT_CODE_RANGE = CodeRange(_DEFAULT_POSITION, _DEFAULT_POSITION)
diff --git a/autotyping/guess_type.py b/autotyping/guess_type.py
@@ -1,4 +1,4 @@
-from typing import Tuple, Optional, List, TYPE_CHECKING
+from typing import Tuple, Optional, List
 import re
 
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from typing import Tuple, Optional, List, TYPE_CHECKING`
	`1`	`+from typing import Tuple, Optional, List`
`2`	`2`	`import re`
`3`	`3`
`4`	`4`