diff --git a/docs/task_on_kart.rst b/docs/task_on_kart.rst index 69702df4..cc48ac37 100644 --- a/docs/task_on_kart.rst +++ b/docs/task_on_kart.rst @@ -90,6 +90,7 @@ It is also possible to specify a file format other than pkl. The supported file - .feather - .png - .jpg +- .ini If dump something other than the above, can use :func:`~gokart.TaskOnKart.make_model_target`. diff --git a/gokart/file_processor.py b/gokart/file_processor.py index 9dda5658..218d0dfa 100644 --- a/gokart/file_processor.py +++ b/gokart/file_processor.py @@ -269,6 +269,7 @@ def dump(self, obj, file): def make_file_processor(file_path: str, store_index_in_feather: bool) -> FileProcessor: extension2processor = { '.txt': TextFileProcessor(), + '.ini': TextFileProcessor(), '.csv': CsvFileProcessor(sep=','), '.tsv': CsvFileProcessor(sep='\t'), '.pkl': PickleFileProcessor(),