Repository: kornia/kornia

Issue:
spatial_soft_argmax imports cause mypy error
### Describe the bug

Imports in `kornia.geometry.subpix.spatial_soft_argmax` raise error when running mypy.

This is fixed by simply changing the import from:

```python
from kornia.geometry import normalize_pixel_coordinates, normalize_pixel_coordinates3d
```

to:

```python
from kornia.geometry.conversions import normalize_pixel_coordinates, normalize_pixel_coordinates3d
```

### Reproduction steps

When adding kornia imports to a script and running mypy checks with the below settings:

```toml
ignore_missing_imports = true
show_error_codes = true
exclude = "(build|data|dist|logo|logs|output)/"

# Strict
warn_unused_configs = true
disallow_any_generics = true
disallow_subclassing_any = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
disallow_untyped_decorators = true
no_implicit_optional = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_return_any = true
no_implicit_reexport = true
strict_equality = true
```

the following error is encountered:

```bash
mypy .
Traceback (most recent call last):
  File "/home/icorley/miniconda3/envs/torchgeo/bin/mypy", line 8, in <module>
    sys.exit(console_entry())
  File "/home/icorley/miniconda3/envs/torchgeo/lib/python3.9/site-packages/mypy/__main__.py", line 11, in console_entry
    main(None, sys.stdout, sys.stderr)
  File "mypy/main.py", line 87, in main
  File "mypy/main.py", line 165, in run_build
  File "mypy/build.py", line 179, in build
  File "mypy/build.py", line 254, in _build
  File "mypy/build.py", line 2697, in dispatch
  File "mypy/build.py", line 3021, in process_graph
  File "mypy/build.py", line 3138, in process_stale_scc
  File "mypy/build.py", line 2288, in write_cache
  File "mypy/build.py", line 1475, in write_cache
  File "mypy/nodes.py", line 313, in serialize
  File "mypy/nodes.py", line 3149, in serialize
  File "mypy/nodes.py", line 3083, in serialize
AssertionError: Definition of kornia.geometry.subpix.spatial_soft_argmax.normalize_pixel_coordinates is unexpectedly incomplete
```


### Expected behavior

No mypy error raised

### Environment

- PyTorch Version: 1.9.0
- OS: Linux
- How you installed PyTorch: conda
- Build command you used (if compiling from source): N/A
- Python version: 3.9.7
- CUDA/cuDNN version: 11.4
- GPU models and configuration: NVIDIA 1070
- Any other relevant information:


### Additional context

@calebrob6


Discussion:

Resolving patch:
diff --git a/kornia/geometry/subpix/spatial_soft_argmax.py b/kornia/geometry/subpix/spatial_soft_argmax.py
--- a/kornia/geometry/subpix/spatial_soft_argmax.py
+++ b/kornia/geometry/subpix/spatial_soft_argmax.py
@@ -5,7 +5,7 @@
 import torch.nn.functional as F
 
 import kornia
-from kornia.geometry import normalize_pixel_coordinates, normalize_pixel_coordinates3d
+from kornia.geometry.conversions import normalize_pixel_coordinates, normalize_pixel_coordinates3d
 from kornia.geometry.subpix import dsnt
 from kornia.utils import create_meshgrid, create_meshgrid3d
 from kornia.utils.helpers import _torch_solve_cast


Test patch: