Support for protocols / runtime_checkable breaks

[vwxyzjn]

from dataclasses import dataclass
from typing import Protocol, Optional, runtime_checkable

import pydantic
import tyro

@runtime_checkable
class ClassA(Protocol):
    def test(self) -> float:
        ...

@dataclass
class ClassB(pydantic.BaseModel):
    class_a: ClassA
    def test(self) -> float:
        return self.class_a.test()

@dataclass
class Args:
    class_b: ClassB

if __name__ == "__main__":
    args = tyro.cli(Args)
    print(args)

[brentyi]

Interesting! Are you sure the error there is a tyro bug?

I can get the same error even after removing the tyro and dataclass parts:

from typing import Protocol, runtime_checkable

import pydantic


@runtime_checkable
class ClassA(Protocol):
    def test(self) -> float: ...


class ClassB(pydantic.BaseModel):
    class_a: ClassA

    def test(self) -> float:
        return self.class_a.test()