Skip to content

Make ts.keys receive a torchstore name as arguments, fix ts.initialize type annotations#125

Open
dvmazur wants to merge 2 commits intometa-pytorch:mainfrom
dvmazur:main
Open

Make ts.keys receive a torchstore name as arguments, fix ts.initialize type annotations#125
dvmazur wants to merge 2 commits intometa-pytorch:mainfrom
dvmazur:main

Conversation

@dvmazur
Copy link

@dvmazur dvmazur commented Feb 10, 2026

This PR does the following:

  • FIxes type annotation for ts.initialize (HostMesh -> ProcMesh)
  • Adds a store_name argument to ts.keys

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Feb 10, 2026
@dvmazur dvmazur changed the title Make ts.keys receive a torchstore name as arguments Make ts.keys receive a torchstore name as arguments, fix ts.initialize type annotations Feb 11, 2026
Copy link
Contributor

@LucasLLC LucasLLC left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

tysm!

@dvmazur
Copy link
Author

dvmazur commented Feb 11, 2026

I don't have a merge button by the way. What's the procedure here? :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants