fix annotating __init__.py files by stripping __init__ from modpath

Author: karimkhaleel

auto_type_annotate.py

@@ -36,6 +36,7 @@ def _to_mod(fname: str, roots: tuple[str, ...]) -> str:
             relative.removesuffix('.py')
             .replace('/', '.')
             .replace('\\', '.')
+            .removesuffix('.__init__')
         )
     else:
         raise AssertionError(f'{fname=} not found in {roots=}')

tests/auto_type_annotate_test.py

@@ -34,6 +34,11 @@ def test_to_mod_src_layout():
     assert _to_mod('src/foo.py', ('.', 'src')) == 'foo'
     assert _to_mod('bar.py', ('.', 'src')) == 'bar'
 
+def test_to_mod_init_files():
+    assert _to_mod('a/__init__.py', ('.',)) == 'a'
+    assert _to_mod('a/__init__b.py', ('.',)) == 'a.__init__b'
+    assert _to_mod('a/__init__b/c.py', ('.',)) == 'a.__init__b.c'
+
 
 def _find_untyped(s):
     visitor = FindUntyped()