Make validate private by renaming validate to _validate (pytorch#107927)

Pull Request resolved: pytorch#107927 Approved by: https://github.com/tugsbayasgalan
rui-ren · Aug 25, 2023 · 39854df · 39854df
1 parent f2f8285
commit 39854df
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 8 deletions.
diff --git a/docs/source/export.rst b/docs/source/export.rst
@@ -17,7 +17,6 @@ torch.export
 .. autoclass:: ExportedProgram
 
     .. automethod:: transform
-    .. automethod:: validate
     .. automethod:: module
 
 .. autoclass:: ExportBackwardSignature

diff --git a/test/export/test_export.py b/test/export/test_export.py
@@ -143,7 +143,7 @@ def forward(self, x, y):
             {},
             preserve_module_call_signature=("foo.nested", "foo"),
         )
-        ep.validate()
+        ep._validate()
         self.assertEqual(len(ep.module_call_graph), 2)
         # TODO(zhxchen17) unflattener
         # unflattened = unflatten(export_module)

diff --git a/torch/export/__init__.py b/torch/export/__init__.py
@@ -558,12 +558,7 @@ def _check_input_constraints(self, *args):
         _assertion_graph = _assertion_graph_res.graph_module
         _assertion_graph(*args)
 
-    def validate(self):
-        """
-        .. warning::
-            Do not use.
-
-        """
+    def _validate(self):
         # TODO(zhxchen17) check for get_attr
         # TODO(zhxchen17) check for funcitonal ops
         for gm in self.graph_module.modules():