From cb7e162230d6ca3849a183435d05c6f802498de1 Mon Sep 17 00:00:00 2001 From: Niccolo-Ajroldi Date: Thu, 3 Oct 2024 19:03:42 +0200 Subject: [PATCH] fix isort --- .../external_tuning/pytorch_nadamw_full_budget.py | 2 +- .../external_tuning/pytorch_nadamw_target_setting.py | 2 +- .../self_tuning/pytorch_nadamw_full_budget.py | 2 +- .../self_tuning/pytorch_nadamw_target_setting.py | 2 +- .../development_algorithms/cifar/cifar_jax/submission.py | 2 +- .../development_algorithms/cifar/cifar_pytorch/submission.py | 2 +- .../development_algorithms/mnist/mnist_jax/submission.py | 2 +- .../development_algorithms/mnist/mnist_pytorch/submission.py | 2 +- .../paper_baselines/adafactor/jax/submission.py | 2 +- .../paper_baselines/adafactor/pytorch/submission.py | 2 +- reference_algorithms/paper_baselines/adamw/jax/submission.py | 2 +- .../paper_baselines/adamw/pytorch/submission.py | 2 +- reference_algorithms/paper_baselines/lamb/jax/submission.py | 2 +- reference_algorithms/paper_baselines/lamb/pytorch/submission.py | 2 +- .../paper_baselines/nadamw/pytorch/submission.py | 2 +- reference_algorithms/paper_baselines/shampoo/jax/submission.py | 2 +- submissions/template/submission.py | 2 +- 17 files changed, 17 insertions(+), 17 deletions(-) diff --git a/prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py b/prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py index c85cc6dd3..72a3bf289 100644 --- a/prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py +++ b/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 diff --git a/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py b/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py index bb1278911..934538b63 100644 --- a/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py +++ b/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 diff --git a/prize_qualification_baselines/self_tuning/pytorch_nadamw_full_budget.py b/prize_qualification_baselines/self_tuning/pytorch_nadamw_full_budget.py index 2af6d548a..f968d4abf 100644 --- a/prize_qualification_baselines/self_tuning/pytorch_nadamw_full_budget.py +++ b/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 diff --git a/prize_qualification_baselines/self_tuning/pytorch_nadamw_target_setting.py b/prize_qualification_baselines/self_tuning/pytorch_nadamw_target_setting.py index 2e2385e29..14c22141c 100644 --- a/prize_qualification_baselines/self_tuning/pytorch_nadamw_target_setting.py +++ b/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 diff --git a/reference_algorithms/development_algorithms/cifar/cifar_jax/submission.py b/reference_algorithms/development_algorithms/cifar/cifar_jax/submission.py index 89aeac238..7e41e9fd7 100644 --- a/reference_algorithms/development_algorithms/cifar/cifar_jax/submission.py +++ b/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 diff --git a/reference_algorithms/development_algorithms/cifar/cifar_pytorch/submission.py b/reference_algorithms/development_algorithms/cifar/cifar_pytorch/submission.py index bcdab6fc3..81110bae6 100644 --- a/reference_algorithms/development_algorithms/cifar/cifar_pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/development_algorithms/mnist/mnist_jax/submission.py b/reference_algorithms/development_algorithms/mnist/mnist_jax/submission.py index 01a266eaf..3f75c9904 100644 --- a/reference_algorithms/development_algorithms/mnist/mnist_jax/submission.py +++ b/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 diff --git a/reference_algorithms/development_algorithms/mnist/mnist_pytorch/submission.py b/reference_algorithms/development_algorithms/mnist/mnist_pytorch/submission.py index e72a9d823..d326f4035 100644 --- a/reference_algorithms/development_algorithms/mnist/mnist_pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/adafactor/jax/submission.py b/reference_algorithms/paper_baselines/adafactor/jax/submission.py index ea440cce7..39cf3d4f9 100644 --- a/reference_algorithms/paper_baselines/adafactor/jax/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/adafactor/pytorch/submission.py b/reference_algorithms/paper_baselines/adafactor/pytorch/submission.py index 30d6942e4..880f9168d 100644 --- a/reference_algorithms/paper_baselines/adafactor/pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/adamw/jax/submission.py b/reference_algorithms/paper_baselines/adamw/jax/submission.py index 935d0d0ca..06eeacb39 100644 --- a/reference_algorithms/paper_baselines/adamw/jax/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/adamw/pytorch/submission.py b/reference_algorithms/paper_baselines/adamw/pytorch/submission.py index ddd17b3b2..0710fb9a0 100644 --- a/reference_algorithms/paper_baselines/adamw/pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/lamb/jax/submission.py b/reference_algorithms/paper_baselines/lamb/jax/submission.py index 3944d6483..891da63be 100644 --- a/reference_algorithms/paper_baselines/lamb/jax/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/lamb/pytorch/submission.py b/reference_algorithms/paper_baselines/lamb/pytorch/submission.py index 20bc80d23..7886dc75d 100644 --- a/reference_algorithms/paper_baselines/lamb/pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/nadamw/pytorch/submission.py b/reference_algorithms/paper_baselines/nadamw/pytorch/submission.py index c85cc6dd3..72a3bf289 100644 --- a/reference_algorithms/paper_baselines/nadamw/pytorch/submission.py +++ b/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 diff --git a/reference_algorithms/paper_baselines/shampoo/jax/submission.py b/reference_algorithms/paper_baselines/shampoo/jax/submission.py index 104ae3ce3..e853a821b 100644 --- a/reference_algorithms/paper_baselines/shampoo/jax/submission.py +++ b/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 diff --git a/submissions/template/submission.py b/submissions/template/submission.py index fb9b1cad1..9bfb23367 100644 --- a/submissions/template/submission.py +++ b/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