Avoid partially initialized module ImportError

Question

Context

Let\'s say I have the following module:

.
└── my_module
    ├── __init__.py
    ├── my_module_A.py (defines ClassA)
    └── my_module_B.py (defin