revert import order changes

Author: priyakasimbeg

algoperf/workloads/imagenet_resnet/imagenet_jax/randaugment.py

@@ -7,14 +7,13 @@
 import inspect
 import math
 
-import tensorflow as tf
-
 from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     rotate_img
 from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     transform
 from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     translate
+import tensorflow as tf
 
 # This signifies the max integer that the controller RNN could predict for the
 # augmentation scheme.

algoperf/workloads/imagenet_vit/workload.py

@@ -3,7 +3,8 @@
 from typing import Dict, Iterator, Optional
 
 from algoperf import spec
-from algoperf.workloads.imagenet_resnet.workload import BaseImagenetResNetWorkload
+from algoperf.workloads.imagenet_resnet.workload import \
+    BaseImagenetResNetWorkload
 
 
 def decode_variant(variant: str) -> Dict[str, int]:

algoperf/workloads/librispeech_deepspeech/librispeech_jax/workload.py

@@ -1,15 +1,15 @@
 import functools
 from typing import Dict, Optional, Tuple
 
+from flax import jax_utils
 import jax
 import jax.numpy as jnp
 import numpy as np
-from flax import jax_utils
 
-from algoperf import param_utils, spec
-from algoperf.workloads.librispeech_conformer.librispeech_jax.workload import (
-  LibriSpeechConformerWorkload,
-)
+from algoperf import param_utils
+from algoperf import spec
+from algoperf.workloads.librispeech_conformer.librispeech_jax.workload import \
+    LibriSpeechConformerWorkload
 from algoperf.workloads.librispeech_deepspeech.librispeech_jax import models
 
 

algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py

@@ -3,18 +3,17 @@
 import torch
 from torch.nn.parallel import DistributedDataParallel as DDP
 
-from algoperf import param_utils, spec
+from algoperf import param_utils
+from algoperf import spec
 from algoperf.pytorch_utils import pytorch_setup
-from algoperf.workloads.librispeech_conformer.librispeech_pytorch.models import (
-  initialize,
-)
-from algoperf.workloads.librispeech_conformer.librispeech_pytorch.workload import (
-  LibriSpeechConformerWorkload,
-)
-from algoperf.workloads.librispeech_deepspeech.librispeech_pytorch.models import (
-  DeepspeechConfig,
-  DeepspeechEncoderDecoder,
-)
+from algoperf.workloads.librispeech_conformer.librispeech_pytorch.models import \
+    initialize
+from algoperf.workloads.librispeech_conformer.librispeech_pytorch.workload import \
+    LibriSpeechConformerWorkload
+from algoperf.workloads.librispeech_deepspeech.librispeech_pytorch.models import \
+    DeepspeechConfig
+from algoperf.workloads.librispeech_deepspeech.librispeech_pytorch.models import \
+    DeepspeechEncoderDecoder
 
 USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_setup()
 

algoperf/workloads/ogbg/workload.py

@@ -9,7 +9,8 @@
 
 from algoperf import random_utils as prng
 from algoperf import spec
-from algoperf.workloads.ogbg import input_pipeline, metrics
+from algoperf.workloads.ogbg import input_pipeline
+from algoperf.workloads.ogbg import metrics
 
 
 class BaseOgbgWorkload(spec.Workload):