Fix linting

Author: krammnic

torchtune/datasets/_alpaca.py

@@ -6,7 +6,7 @@
 
 from functools import partial
 
-from typing import Any, Dict, Optional, Union, Callable 
+from typing import Any, Callable, Dict, Optional, Union
 
 from torchtune.data._messages import AlpacaToMessages
 

torchtune/datasets/multimodal/_llava_instruct.py

@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 from pathlib import Path
-from typing import Any, Dict, Optional, Callable
+from typing import Any, Callable, Dict, Optional
 
 from torchtune.data._messages import ShareGPTToMessages
 from torchtune.datasets._sft import SFTDataset

torchtune/datasets/multimodal/_multimodal.py

@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 from pathlib import Path
-from typing import Any, Dict, Optional, Callable
+from typing import Any, Callable, Dict, Optional
 
 from torchtune.data._messages import ShareGPTToMessages
 from torchtune.datasets._sft import SFTDataset

torchtune/datasets/multimodal/_the_cauldron.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-from typing import Any, Dict, Mapping, Optional, Callable
+from typing import Any, Callable, Dict, Mapping, Optional
 
 from torchtune.data._messages import Message
 from torchtune.datasets._sft import SFTDataset