This commit is contained in:
Bouke Versteegh 2020-05-21 22:55:26 +02:00 committed by GitHub
parent 499489f1d3
commit 7e9ba0866c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -75,8 +75,7 @@ def get_ref_type(package: str, imports: set, type_name: str, unwrap: bool = True
if type_name == "google.protobuf.Timestamp":
return "datetime"
else:
if wrapper:
elif wrapper:
imports.add(f"from {wrapper.__module__} import {wrapper.__name__}")
return f"{wrapper.__name__}"