Pylance报reportAttributeAccessIssue错误的原因及解决咨询

代码定义 #

Process协议类 #

from typing import Protocol, runtime_checkable
import torch.nn as nn

@runtime_checkable
class Process(Protocol):
    network: nn.Module

    def update(self, **kwargs):
        ...

ProcessRunner类 #

from typing import List
import torch.nn as nn

class ProcessRunner:
    def __init__(self, network: nn.Module) -> None:
        self.network: nn.Module = network

    def run(self, processes: List[Process]):
        for process in processes:
            process.network = self.network
            self.network = process.update()

Pylance错误信息 #

Cannot assign member "network" for type "Process"
Type "Module | None" cannot be assigned to type "Module"
"None" is incompatible with "Module"PylancereportAttributeAccessIssue

Cannot assign member "network" for type "ProcessRunner*"
"None" is incompatible with "Module"PylancereportAttributeAccessIssue

疑问点 #

• 第二个错误是否因update方法未返回network？若仅在ProcessRunner类中保留network，Process的update方法该如何访问它？
• 第一个错误中，Process协议和ProcessRunner类的network均为nn.Module类型，为何仍触发错误？是否由第二个错误导致？

错误原因与解决方法 #

错误根源 #

两个错误的核心都是Process协议的update方法未明确标注返回类型：Pylance对未标注返回类型的函数，默认会推断其返回值可能包含None，这就和代码中明确声明的nn.Module类型产生冲突。

针对疑问的解答 #

1. 第二个错误的原因与替代方案
   第二个错误确实是因为update没有明确返回nn.Module类型，Pylance认为它可能返回None，而self.network被声明为nn.Module，无法接受None值，因此触发错误。
   如果想只在ProcessRunner中保留network，可以修改Process协议的update方法，让它接收network作为参数，无需通过属性访问：

   @runtime_checkable
   class Process(Protocol):
       def update(self, network: nn.Module, **kwargs) -> nn.Module:
           ...
   

   对应的ProcessRunner.run方法也要调整：

   def run(self, processes: List[Process]):
       for process in processes:
           self.network = process.update(self.network)
   

2. 第一个错误的原因
   第一个错误并非类型本身不匹配，而是连锁反应导致的：因为process.update()可能返回None，Pylance会推断self.network的类型变为nn.Module | None，当你把这个可能为None的值赋值给process.network（要求必须是nn.Module）时，就触发了类型不兼容的错误。根源还是update方法的返回类型未明确标注。

最简修复方案 #

直接给Process协议的update方法添加明确的返回类型标注，确保Pylance知道它返回的是nn.Module：

@runtime_checkable
class Process(Protocol):
    network: nn.Module

    def update(self, **kwargs) -> nn.Module:
        ...

这样两个错误都会消失，因为Pylance可以确认update的返回值类型符合要求，不会出现None的情况。

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