diff --git a/pyproject.toml b/pyproject.toml index 99fc01008fa..845eb6687db 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -106,7 +106,9 @@ select = [ "PT025", "PT026", "PYI", + "RUF008", # mutable dataclass default "RUF015", # access first ele in constant time + "RUF016", # type error non-integer index "RUF017", "TRY200", "TRY302", diff --git a/torch/distributed/fsdp/_trace_utils.py b/torch/distributed/fsdp/_trace_utils.py index 42d569f1ecb..c768b73b8f9 100644 --- a/torch/distributed/fsdp/_trace_utils.py +++ b/torch/distributed/fsdp/_trace_utils.py @@ -1,6 +1,6 @@ import functools from contextlib import contextmanager -from dataclasses import dataclass +from dataclasses import dataclass, field from typing import Any, Callable, Dict, List, NamedTuple, Optional, Set, Tuple import torch @@ -27,7 +27,7 @@ class TracingConfig: in :meth:`~torch.fx.Tracer.trace`. """ - tracer: torch.fx.Tracer = torch.fx.Tracer() + tracer: torch.fx.Tracer = field(default_factory=torch.fx.Tracer) concrete_args: Optional[Dict[str, Any]] = None