包 __init__.py 导入所有子文件，但只从另一个脚本加载一个子文件？

我创建了一个具有以下文件结构的包：

- package
  - __init__.py
  - load.py
  - train.py
  - test.py

My __init__.pyfile 只是这些文件的类的导入：

from package.load import Load
from package.train import Train
from package.test import Test

大多数时候，我想加载所有三个类，但有时我只想专门加载这些类之一。例如，在临时脚本（包外）中，我希望能够仅调用 Load 类，如下所示：

from package import Load

虽然上述所有内容都适用于此设计，但我遇到一个问题，即在导入时也会加载来自训练/测试的依赖项Load就像上面的那样。我该如何设置__init__.py文件，这样我就可以做同样的事情import调用而不获取从训练/测试加载的依赖项？

补充说明：

为什么我这样做：我有一个问题，我希望一些人能够使用Load类，它仅使用基本 python，但是训练/测试文件包含专门的依赖项，仅 Load 类的用户不想使用甚至安装这些依赖项。

这是一种做一些非常接近你想要的事情的方法。而不是无条件地importing 中的所有包的类__init__.py，您可以在其中定义一个函数来显式导入任何所需的函数（如果未指定，则导入所有函数）。

__init__.py:

from pathlib import Path
import sys

print(f'In {Path(__file__).name}')

package_name = Path(__file__).parent.name
package_prefix = package_name + '.'
class_to_module_map = {'Load': 'load', 'Train': 'train', 'Test': 'test'}


def import_classes(*class_names):
    namespace = sys._getframe(1).f_globals  # Caller's globals.

    if not class_names:
        class_names = class_to_module_map.keys()  # Import them all.

    for class_name in class_names:
        module = class_to_module_map[class_name]
        temp = __import__(package_prefix+module, globals(), locals(), [class_name])
        namespace[class_name] = getattr(temp, class_name)  # Add to caller's namespace.

出于测试目的，这是我放入的内容load.py script:
（我还在其他两个模块中添加了类似的内容，以验证它们是否得到了imported.)

load.py:

from pathlib import Path

print(f'In {Path(__file__).name}')

class Load: pass

最后，这是一个仅使用它的示例import the Load class:

ad_hoc.py:

from my_package import import_classes

#from my_package import Load
import_classes('Load')

test = Load()
print(test)

随着输出的产生：

In __init__.py
In load.py
<my_package.load.Load object at 0x001FE4A8>