docs tweaks

determined-ai · Mar 26, 2024 · 9806379 · 9806379
1 parent 43d080f
commit 9806379
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/docs/reference/training/api-core-reference.rst b/docs/reference/training/api-core-reference.rst
@@ -68,6 +68,14 @@
    :members:
    :member-order: bysource
 
+*************************************
+ ``determined.core.ProfilerContext``
+*************************************
+
+.. autoclass:: determined.core.ProfilerContext
+   :members:
+   :member-order: bysource
+
 ***************************************
  ``determined.core.SearcherOperation``
 ***************************************

diff --git a/harness/determined/core/_profiler.py b/harness/determined/core/_profiler.py
@@ -32,6 +32,7 @@ class ProfilerContext:
     turned on/off as such:
 
     .. code::
+
         with det.core.init() as core_context:
             core_context.profiler.on()
             ...