fix mypy return

gene-db · Apr 8, 2024 · 62277ac · 62277ac
1 parent 807d574
commit 62277ac
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/pyspark/sql/types.py b/python/pyspark/sql/types.py
@@ -1348,9 +1348,10 @@ class VariantType(AtomicType):
     def needConversion(self) -> bool:
         return True
 
-    def fromInternal(self, obj: Dict) -> "VariantVal":
-        if obj is not None and all(key in obj for key in ["value", "metadata"]):
-            return VariantVal(obj["value"], obj["metadata"])
+    def fromInternal(self, obj: Dict) -> Optional["VariantVal"]:
+        if obj is None or all(key in obj for key in ["value", "metadata"]):
+            return None
+        return VariantVal(obj["value"], obj["metadata"])
 
 
 class UserDefinedType(DataType):