fix isort

prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py

@@ -1,7 +1,7 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py

@@ -1,7 +1,7 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

prize_qualification_baselines/self_tuning/pytorch_nadamw_full_budget.py

@@ -1,7 +1,7 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

prize_qualification_baselines/self_tuning/pytorch_nadamw_target_setting.py

@@ -1,7 +1,7 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

reference_algorithms/development_algorithms/cifar/cifar_jax/submission.py

@@ -1,7 +1,7 @@
 """Training algorithm track submission functions for CIFAR10."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

reference_algorithms/development_algorithms/cifar/cifar_pytorch/submission.py

@@ -1,6 +1,6 @@
 """Training algorithm track submission functions for CIFAR10."""
 
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 import torch
 from torch.optim.lr_scheduler import CosineAnnealingLR

reference_algorithms/development_algorithms/mnist/mnist_jax/submission.py

@@ -1,7 +1,7 @@
 """Training algorithm track submission functions for MNIST."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

reference_algorithms/development_algorithms/mnist/mnist_pytorch/submission.py

@@ -1,6 +1,6 @@
 """Training algorithm track submission functions for MNIST."""
 
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 import torch
 

reference_algorithms/paper_baselines/adafactor/jax/submission.py

@@ -1,7 +1,7 @@
 """Submission file for an Adafactor optimizer with warmup+cosine LR in Jax."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

reference_algorithms/paper_baselines/adafactor/pytorch/submission.py

@@ -1,7 +1,7 @@
 """Submission file for Adafactor in PyTorch."""
 
 from functools import partial
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

reference_algorithms/paper_baselines/adamw/jax/submission.py

@@ -1,7 +1,7 @@
 """Submission file for an AdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

reference_algorithms/paper_baselines/adamw/pytorch/submission.py

@@ -1,6 +1,6 @@
 """Submission file for an AdamW optimizer with warmup+cosine LR in PyTorch."""
 
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

reference_algorithms/paper_baselines/lamb/jax/submission.py

@@ -1,7 +1,7 @@
 """Submission file for a LAMB optimizer with warmup+cosine LR in Jax."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

reference_algorithms/paper_baselines/lamb/pytorch/submission.py

@@ -1,7 +1,7 @@
 """Submission file for a LAMB optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

reference_algorithms/paper_baselines/nadamw/pytorch/submission.py

@@ -1,7 +1,7 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in PyTorch."""
 
 import math
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from absl import logging
 import torch

reference_algorithms/paper_baselines/shampoo/jax/submission.py

@@ -1,7 +1,7 @@
 """Submission file for a Shampoo optimizer with warmup+cosine LR in Jax."""
 
 import functools
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from flax import jax_utils
 import jax

submissions/template/submission.py

@@ -4,7 +4,7 @@
 and https://github.com/mlcommons/algorithmic-efficiency/blob/main/DOCUMENTATION.md#disallowed-submissions
 for guidelines.
 """
-from typing import Dict, Iterator, List, Tuple, Any
+from typing import Any, Dict, Iterator, List, Tuple
 
 from algorithmic_efficiency import spec