Mypy показывает ошибку для функции замены pyspark ⇐ Python

[Anonymous]

Mypy показывает ошибку для функции pyspark replace.

Код: Выделить всё

main.py

:

Код: Выделить всё

from pyspark.sql import SparkSession
from pyspark.sql.functions import *

spark = SparkSession.builder.appName(__name__).getOrCreate()
x = {'a': 'b', 'c': 'd'}
df = spark.read.parquet('abc')
df.replace(x, subset=df.columns)

Сообщение об ошибке:

Код: Выделить всё

$ mypy main.py
main.py:7: error: Argument 1 to "replace" of "DataFrame" has incompatible type "dict[str, str]"; expected "dict[bool | float | int | str, bool | float | int | str | None]"  [arg-type]
Found 1 error in 1 file (checked 1 source file)

Любая помощь приветствуется.

Подробнее здесь: https://stackoverflow.com/questions/778 ... e-function