We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9153ab7 commit 57d07e1Copy full SHA for 57d07e1
jax/profiler.py
@@ -14,6 +14,7 @@
14
15
# Note: import <name> as <name> is required for names to be exported.
16
# See PEP 484 & https://github.com/jax-ml/jax/issues/7570
17
+from typing import Any
18
19
from jax._src.profiler import (
20
ProfileOptions as ProfileOptions,
@@ -28,3 +29,9 @@
28
29
stop_trace as stop_trace,
30
trace as trace,
31
)
32
+
33
+# this is a temporary shim to please pytype in the meantime before the migration
34
+# is complete for cl/760646494
35
+ProfileData: Any = None
36
+ProfileEvent: Any = None
37
+ProfilePlane: Any = None
0 commit comments