|
@@ -188,7 +188,7 @@ class CSVDataNode(DataNode, _FileDataNodeMixin, _TabularDataNodeMixin):
|
|
|
)
|
|
|
else:
|
|
|
self._convert_data_to_dataframe(exposed_type, data).to_csv(
|
|
|
- self._path, index=False, encoding=self.properties[self.__ENCODING_KEY], header=None
|
|
|
+ self._path, index=False, encoding=self.properties[self.__ENCODING_KEY], header=False
|
|
|
)
|
|
|
|
|
|
def write_with_column_names(self, data: Any, columns: Optional[List[str]] = None, job_id: Optional[JobId] = None):
|
|
@@ -201,6 +201,6 @@ class CSVDataNode(DataNode, _FileDataNodeMixin, _TabularDataNodeMixin):
|
|
|
"""
|
|
|
df = self._convert_data_to_dataframe(self.properties[self._EXPOSED_TYPE_PROPERTY], data)
|
|
|
if columns and isinstance(df, pd.DataFrame):
|
|
|
- df.columns = columns
|
|
|
+ df.columns = pd.Index(columns, dtype="object")
|
|
|
df.to_csv(self._path, index=False, encoding=self.properties[self.__ENCODING_KEY])
|
|
|
self.track_edit(timestamp=datetime.now(), job_id=job_id)
|