Skip to content

Commit

Permalink
fix lint (pytorch#119395)
Browse files Browse the repository at this point in the history
Summary: as title

Test Plan: lint

Differential Revision: D53532399

Pull Request resolved: pytorch#119395
Approved by: https://github.com/tugsbayasgalan, https://github.com/malfet
  • Loading branch information
suo authored and pytorchmergebot committed Feb 7, 2024
1 parent 23b030a commit 0e2330d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch/_export/serde/serialize.py
Original file line number Diff line number Diff line change
@@ -1474,9 +1474,9 @@ def deserialize_torch_artifact(self, serialized: bytes):
break

assert spec is not None
fake_tensor = self.serialized_name_to_meta[spec.arg.name]
fake_tensor = self.serialized_name_to_meta[spec.arg.name] # type: ignore[union-attr]
if spec.kind == ep.InputKind.PARAMETER:
fake_tensor = torch.nn.Parameter(fake_tensor)
fake_tensor = torch.nn.Parameter(fake_tensor) # type: ignore[assignment,arg-type]
artifact[name] = fake_tensor
return artifact

0 comments on commit 0e2330d

Please sign in to comment.