From e3c523c98cbc9ecdd37e36da19848ca9d0aef4a3 Mon Sep 17 00:00:00 2001
From: Ned Batchelder <ned@nedbatchelder.com>
Date: Sat, 31 Dec 2022 19:04:31 -0500
Subject: mypy: add __init__.py, __main__.py, bytecode.py, context.py,
 exceptions.py, report.py, and version.py

---
 coverage/bytecode.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'coverage/bytecode.py')

diff --git a/coverage/bytecode.py b/coverage/bytecode.py
index ceb18cf3..15bf755b 100644
--- a/coverage/bytecode.py
+++ b/coverage/bytecode.py
@@ -3,10 +3,11 @@
 
 """Bytecode manipulation for coverage.py"""
 
-import types
+from types import CodeType
+from typing import Generator
 
 
-def code_objects(code):
+def code_objects(code: CodeType) -> Generator[CodeType, None, None]:
     """Iterate over all the code objects in `code`."""
     stack = [code]
     while stack:
@@ -14,6 +15,6 @@ def code_objects(code):
         # push its children for later returning.
         code = stack.pop()
         for c in code.co_consts:
-            if isinstance(c, types.CodeType):
+            if isinstance(c, CodeType):
                 stack.append(c)
         yield code
-- 
cgit v1.2.1