Add exception handling in docs

pytorch · May 18, 2024 · 6cb086a · 6cb086a
1 parent 1fc1e6a
commit 6cb086a
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 10 deletions.
diff --git a/fbgemm_gpu/experimental/gen_ai/gen_ai/__init__.py b/fbgemm_gpu/experimental/gen_ai/gen_ai/__init__.py
@@ -16,9 +16,9 @@
     # @manual=//deeplearning/fbgemm/fbgemm_gpu:test_utils
     from fbgemm_gpu import open_source
 
-    # # pyre-ignore[21]
-    # # @manual=//deeplearning/fbgemm/fbgemm_gpu:test_utils
-    # from fbgemm_gpu.docs.version import __version__  # noqa: F401
+    # pyre-ignore[21]
+    # @manual=//deeplearning/fbgemm/fbgemm_gpu:test_utils
+    from fbgemm_gpu.docs.version import __version__  # noqa: F401
 except Exception:
     open_source: bool = False
 

diff --git a/fbgemm_gpu/fbgemm_gpu/__init__.py b/fbgemm_gpu/fbgemm_gpu/__init__.py
@@ -19,13 +19,13 @@
 # and use its existence to determine whether or not we are in OSS context
 open_source: bool = True
 
+# Trigger the manual addition of docstrings to pybind11-generated operators
+import fbgemm_gpu.docs  # noqa: F401, E402
+
+# Export the version string from the version file auto-generated by setup.py
+from fbgemm_gpu.docs.version import __version__  # noqa: F401, E402
+
 try:
-    # Trigger the manual addition of docstrings to pybind11-generated operators
-    import fbgemm_gpu.docs  # noqa: F401, E402
-
-    # Export the version string from the version file auto-generated by setup.py
-    from fbgemm_gpu.docs.version import __version__  # noqa: F401, E402
-
     # Trigger meta operator registrations
     from . import sparse_ops  # noqa: F401, E402
 except Exception as e:

diff --git a/fbgemm_gpu/fbgemm_gpu/docs/__init__.py b/fbgemm_gpu/fbgemm_gpu/docs/__init__.py
@@ -6,4 +6,7 @@
 # LICENSE file in the root directory of this source tree.
 
 # Trigger the manual addition of docstrings to pybind11-generated operators
-from . import jagged_tensor_ops, table_batched_embedding_ops  # noqa: F401
+try:
+    from . import jagged_tensor_ops, table_batched_embedding_ops  # noqa: F401
+except Exception as e:
+    pass