cleanup

2020-05-21 22:55:26 +02:00
parent 499489f1d3
commit 7e9ba0866c
1 changed files with 3 additions and 4 deletions
--- a/betterproto/plugin.py
+++ b/betterproto/plugin.py
@@ -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__}"