Fix lint (#347)

koxudaxi · Feb 18, 2021 · 6f706f0 · 6f706f0
1 parent 4e04f11
commit 6f706f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datamodel_code_generator/parser/jsonschema.py b/datamodel_code_generator/parser/jsonschema.py
@@ -749,9 +749,9 @@ def parse_root_type(
         additional_properties: Optional[JsonSchemaObject] = None,
     ) -> DataType:
         if obj.ref:
-            data_type = self.get_ref_data_type(obj.ref)
+            data_type: DataType = self.get_ref_data_type(obj.ref)
         elif obj.type:
-            data_type: DataType = self.get_data_type(obj)
+            data_type = self.get_data_type(obj)
         elif obj.anyOf:
             data_type = self.parse_any_of(name, obj, [*path, name])
         elif obj.oneOf: