适用于目录和软件包的Python导入

我有两个类(在同一目录中),一个类从另一个派生。

ClasssA.py包含以下代码:

class ClassA():

    def __init__(self):
        pass

ClasssB.py包含以下代码:

from ClassA import ClassA

class ClassB(ClassA):

    def __init__(self):

        ClassA.__init__(self)

当我从Python控制台运行脚本时,这可以正常工作。 我想将它们构建到包中,但是当我尝试实例化ClassB时,出现错误:

没有名为“类”的模块

How can I write the import statement and/or __init__.py so the code works in both scenarios.

感谢您的任何建议。