Tristan Rice dc7d8a889e caffe2: refactor context to allow being typed (#48340) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48340 This changes the context managed classes from using a decorator to define them to using inheritance. Inheritance allows the python static type checking to work correctly. ``` context.define_context() class Bar(object): ... context.define_context(allow_default=True) class Foo(object): ... ``` becomes ``` class Foo(context.Managed): ... class Bar(context.DefaultManaged): ... ``` Behavior differences: * arg_name has been removed since it's not used anywhere * classes need to call `super()` in `__enter__/__exit__` methods if they override (none do) This also defines a context.pyi file to add types for python3. python2 support should not be affected Test Plan: ci buck test //caffe2/caffe2/python:context_test //caffe2/caffe2/python:checkpoint_test Reviewed By: dongyuzheng Differential Revision: D25133469 fbshipit-source-id: 16368bf723eeb6ce3308d6827f5ac5e955b4e29a		2020-11-30 18:31:14 -08:00
..
__init__.py
adaptive_weight.py
add_bias.py
arc_cosine_feature_map.py
batch_huber_loss.py
batch_lr_loss.py
batch_mse_loss.py
batch_normalization.py
batch_sigmoid_cross_entropy_loss.py
batch_softmax_loss.py
blob_weighted_sum.py
bpr_loss.py
bucket_weighted.py
build_index.py
concat.py
constant_weight.py
conv.py
dropout.py
fc.py
fc_with_bootstrap.py
fc_without_bias.py
feature_sparse_to_dense.py
functional.py
gather_record.py
homotopy_weight.py
label_smooth.py
last_n_window_collector.py	Add last_n_window_collector	2020-10-30 02:35:54 -07:00
layer_normalization.py
layers.py
margin_rank_loss.py
merge_id_lists.py
pairwise_similarity.py
position_weighted.py
random_fourier_features.py
reservoir_sampling.py
sampling_train.py
sampling_trainable_mixin.py
select_record_by_context.py
semi_random_features.py
sparse_dropout_with_replacement.py
sparse_feature_hash.py
sparse_lookup.py
split.py
tags.py	caffe2: refactor context to allow being typed (#48340)	2020-11-30 18:31:14 -08:00
uniform_sampling.py