HyukjinKwon commented on a change in pull request #33954:
URL: https://github.com/apache/spark/pull/33954#discussion_r707863591
##########
File path: python/pyspark/pandas/typedef/typehints.py
##########
@@ -84,11 +87,24 @@ def __repr__(self) -> str:
class DataFrameType(object):
def __init__(
- self, dtypes: List[Dtype], spark_types: List[types.DataType], names:
List[Optional[str]]
+ self,
+ index_name: Optional[str],
+ index_dtype: Dtype,
+ data_dtypes: List[Dtype],
+ data_names: List[Optional[str]],
+ spark_types: List[types.DataType],
):
from pyspark.pandas.internal import InternalField
from pyspark.pandas.utils import name_like_string
+ names = [index_name] + data_names if index_dtype is not None else
data_names
+ dtypes = [index_dtype] + data_dtypes if index_dtype is not None else
data_dtypes
+
+ self.index_name = index_name
+ self.index_dtype = index_dtype
+ self.data_dtypes = data_dtypes
+ self.data_names = data_names
Review comment:
Oh yeah.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
