Matthew Hoffman
|
7a779b5257
|
Add functions from torch.masked._ops to __all__ for torch.masked (#131288)
Add the non-private operations imported in this file to `__all__` so that pyright considers them to be publicly exported. Solves this error:
```
"mean" is not exported from module "torch.masked" Pylance[reportPrivateImportUsage]
```
Related: https://github.com/pytorch/pytorch/pulls?q=pyright+export
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131288
Approved by: https://github.com/ezyang
|
2024-08-01 05:45:08 +00:00 |
|
PyTorch MergeBot
|
91299c95ec
|
Revert "Add functions from torch.masked._ops to __all__ for torch.masked (#131288)"
This reverts commit 78020ea55d.
Reverted https://github.com/pytorch/pytorch/pull/131288 on behalf of https://github.com/kit1980 due to Broke test_public_bindings.py::TestPublicBindings::test_correct_module_names [GH job link](https://github.com/pytorch/pytorch/actions/runs/10172945925/job/28136657243) [HUD commit link](78020ea55d) ([comment](https://github.com/pytorch/pytorch/pull/131288#issuecomment-2259581854))
|
2024-07-31 03:45:09 +00:00 |
|
Matthew Hoffman
|
78020ea55d
|
Add functions from torch.masked._ops to __all__ for torch.masked (#131288)
Add the non-private operations imported in this file to `__all__` so that pyright considers them to be publicly exported. Solves this error:
```
"mean" is not exported from module "torch.masked" Pylance[reportPrivateImportUsage]
```
Related: https://github.com/pytorch/pytorch/pulls?q=pyright+export
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131288
Approved by: https://github.com/ezyang
|
2024-07-31 02:16:38 +00:00 |
|
Xuehai Pan
|
01fc22056a
|
[BE] enable UFMT for torch/masked/ (#127715)
Part of #123062
- #123062
Pull Request resolved: https://github.com/pytorch/pytorch/pull/127715
Approved by: https://github.com/cpuhrsch
|
2024-06-03 22:01:49 +00:00 |
|
George Qi
|
686555b663
|
[maskedtensor] port torch/_masked into torch/masked (#85515)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85515
Approved by: https://github.com/cpuhrsch
|
2022-09-26 23:41:13 +00:00 |
|
George Qi
|
0c46e3ec66
|
[maskedtensor] add basic tests and unary/binary/reduction tests from common_method_invocations (#82841)
Decided offline on the invariant that:
`masked_tensor` calls `MaskedTensor()`, which is analogous to `torch.tensor`
`as_masked_tensor` calls `MaskedTensor._from_values()`, which is analogous to `torch.as_tensor`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82841
Approved by: https://github.com/cpuhrsch, https://github.com/bhosmer
|
2022-09-22 07:37:04 +00:00 |
|
George Qi
|
5e9c26c8e2
|
[maskedtensor] adding reductions (#82839)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82839
Approved by: https://github.com/bhosmer
|
2022-09-06 15:01:35 +00:00 |
|
George Qi
|
94ba085ce0
|
[maskedtensor] first commit, core and creation (#82836)
* __->__ #82836
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82836
Approved by: https://github.com/albanD, https://github.com/bhosmer
|
2022-08-16 20:10:34 +00:00 |
|