如何用Numba jitclass定义含自身类型字段的类？

问题原因 #

你遇到的错误是因为jitclass装饰器处理类定义时，字符串标注的'Foo'还未被解析为Numba可识别的jitclass类型——此时类的编译尚未完成，无法引用自身。而直接替换为object类型会导致Numba无法进行类型优化，且无法正确兼容jitclass实例的赋值。

可行解决方法：使用类型字典+延迟类型绑定 #

通过Numba的类型系统手动定义字段规范，利用占位符实现自引用类型的延迟绑定，同时结合optional类型支持None参数：

from numba.experimental import jitclass
from numba import int64, optional

# 声明类型占位符，用于后续自引用
foo_type = None

# 定义字段类型规范：pred字段使用optional类型，既支持None也支持Foo实例
spec = [
    ('a', int64),
    ('pred', optional(foo_type))
]

@jitclass(spec)
class Foo:
    def __init__(self, pred):
        self.a = 1
        self.pred = pred

# 将编译后的jitclass实例类型赋值给占位符，完成自引用闭环
foo_type = Foo.class_type.instance_type

if __name__ == "__main__":
    # 测试传入None
    x = Foo(None)
    print(x.a)  # 输出 1
    print(x.pred)  # 输出 None
    
    # 测试自引用赋值
    y = Foo(x)
    print(y.pred.a)  # 输出 1

方法说明 #

1. 类型占位符：先定义foo_type作为占位符，在类型规范中引用它，此时Numba会延迟解析该类型。
2. optional类型：optional(foo_type)明确指定字段可以是Foo实例或者None，完美满足你的参数需求。
3. 延迟绑定：类编译完成后，将Foo.class_type.instance_type（即jitclass对应的Numba类型）赋值给占位符，完成自引用类型的绑定。

这种方法不需要外部字典存储字段，直接在jitclass内部实现了自引用类型支持，同时兼容None参数，且能保留Numba的编译优化特性。

内容的提问来源于stack exchange，提问作者Christian