fix: Tensor slicing does not work with ellipsis - Githubissues

synnada-ai / mithril

Mithril: A Modular Machine Learning Library for Model Composability

Apache License 2.0

31 stars 8 forks source link

fix: Tensor slicing does not work with ellipsis #39

Closed aturker-synnada closed 6 days ago

aturker-synnada commented 6 days ago

Description

Closes #38

What is Changed

Scalar now can take EllipsisType as value.

Checklist:

[x] Tests that cover the code added.
[x] Corresponding changes documented.
[x] All tests passed.
[x] The code linted and styled (pre-commit run --all-files has passed).