Skip to content

Commit b45ce37

Browse files
committed
Review notes
1 parent d84768c commit b45ce37

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

mypy/stubgenc.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,7 @@ def generate_module(self) -> None:
395395
for name, obj in items:
396396
if self.is_function(obj):
397397
self.generate_function_stub(name, obj, output=functions)
398-
elif self.is_class(obj):
398+
elif inspect.isclass(obj):
399399
self.generate_class_stub(name, obj, output=types)
400400
else:
401401
self.generate_variable_stub(name, obj, output=variables)
@@ -535,9 +535,6 @@ def is_static_property(self, obj: object) -> bool:
535535
else:
536536
return False
537537

538-
def is_class(self, obj: object) -> bool:
539-
return inspect.isclass(obj) or (self.is_c_module and type(obj) is type(int))
540-
541538
def process_inferred_sigs(self, inferred: list[FunctionSig]) -> None:
542539
for i, sig in enumerate(inferred):
543540
for arg in sig.args:
@@ -769,7 +766,7 @@ def generate_class_stub(self, class_name: str, cls: type, output: list[str]) ->
769766
ro_properties,
770767
class_info,
771768
)
772-
elif self.is_class(value) and self.is_defined_in_module(value):
769+
elif inspect.isclass(value) and self.is_defined_in_module(value):
773770
self.generate_class_stub(attr, value, types)
774771
else:
775772
attrs.append((attr, value))

0 commit comments

Comments
 (0)