Imports missing if PyClasses defined in different modules

[phoerious]

When I have classes in different modules like in the example below and one class makes use of the other, two individual stub files are generated, but don't import each others types.

#[pyclass(subclass, module = "mypkg.mymod_a")]
#[derive(Clone, PartialEq, Eq)]
pub struct MyClass1 {}

#[pyclass(subclass, module = "mypkg.mymod_b")]
#[derive(Clone, PartialEq, Eq)]
pub struct MyClass2 {}

[kstreee-furiosa]

Any progress or workaround for this issue? I'd open to contribute to solve the issue, would anyone can let me know which parts I can investigate to solve the issue?