Delete get_pyobj() entirely (#92638)
Opt for the shorter and more direct node attribute access.
I need to do this because I'm going to publicly document
SymInt and SymFloat but I don't want to doc get_pyobj().
Signed-off-by: Edward Z. Yang <[email protected]>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/92638
Approved by: https://github.com/Chillee, https://github.com/albanD, https://github.com/voznesenskym, https://github.com/bdhirsh
diff --git a/test/test_proxy_tensor.py b/test/test_proxy_tensor.py
index 89a6b45..304e207 100644
--- a/test/test_proxy_tensor.py
+++ b/test/test_proxy_tensor.py
@@ -986,7 +986,7 @@
fx_g = make_fx(f, tracing_mode="symbolic")(torch.randn(5), torch.randn(4))
meta_c = _get_node(fx_g, lambda x: x.target == aten.new_empty.default)
meta_d = _get_node(fx_g, lambda x: x.target == operator.add)
- self.assertTrue(meta_c.meta['val'].shape[0].get_pyobj().expr == meta_d.meta['val'].node.expr)
+ self.assertTrue(meta_c.meta['val'].shape[0].node.expr == meta_d.meta['val'].node.expr)
def test_metadata_fresh(self):
def f(x):
@@ -996,10 +996,10 @@
fx_g = make_fx(f, tracing_mode="symbolic")(torch.randn(3))
meta_cos = _get_node(fx_g, lambda x: x.target == aten.cos.default)
meta_inp = _get_node(fx_g, lambda x: x.op == 'placeholder')
- self.assertTrue(meta_cos.meta['val'].shape[0].get_pyobj().expr == 3)
+ self.assertTrue(meta_cos.meta['val'].shape[0].node.expr == 3)
# Checks if the input expr has been updated even though the constraint
# happened afterwards
- self.assertTrue(meta_inp.meta['val'].shape[0].get_pyobj().expr == 3)
+ self.assertTrue(meta_inp.meta['val'].shape[0].node.expr == 3)
def test_elementwise_meta_with_sym_numbers(self):
def f(x, offset, as_sym_float=False):
