Slice loading errors with integer indices

[hct-synth]

System Info

safetensors==0.4.2

Information

• The official example scripts
• My own modified scripts

Reproduction

from safetensors import safe_open
from safetensors.torch import save_file

import torch

save_file({"test":torch.zeros([10,10])}, "test.st")

with safe_open("test.st", framework="pt", device="cpu") as f:
    print(f.get_slice("test")[0, :])

Expected behavior

I expect this to return a slice of the tensor "test", instead I get the error:

argument 'slices': failed to extract enum Slice ('Slice | Slices')

As far as I can see there's no notes in the documentation or examples about integer slices not working?
https://huggingface.co/docs/safetensors/en/index

[hct-synth]

NB: For anyone running into this, the workaround is to use a slice width of one, which resolves the issue:
print(f.get_slice("test")[0:1, :]) # WORKS

[Narsil]

Thanks a lot for reporting this, I've prepared a fix !

And yes slicing with real slices work, I forgot to add support for real index indexing.
I added support for negative indexing too.

[hct-synth]

@Narsil - thanks for looking into this! Cool to see a PR so soon.