Add context manager to enable/disable profiling (#709)

* add enable_profiling contextmanager

* Update scalene_profiler.py

Fixed spacing.

---------

Co-authored-by: Emery Berger <[email protected]>

README.md

@@ -112,6 +112,13 @@ scalene_profiler.start()
 scalene_profiler.stop()
 ```
 
+```Python
+from scalene.scalene_profiler import enable_profiling
+
+with enable_profiling():
+    # do something
+```
+
 </details>
 
 <details>

scalene/scalene_profiler.py

@@ -67,7 +67,7 @@
     Set,
     Tuple,
     Union,
-    cast,
+    cast, Generator,
 )
 
 import scalene.scalene_config
@@ -147,6 +147,13 @@ def stop() -> None:
     Scalene.stop()
 
 
+@contextlib.contextmanager
+def enable_profiling() -> Generator[None, None, None]:
+    """Contextmanager that starts and stops profiling"""
+    start()
+    yield
+    stop()
+
 class Scalene:
     """The Scalene profiler itself."""