Remove unnecessary isort=off commands

Author: fsschneider

prize_qualification_baselines/external_tuning/jax_nadamw_full_budget.py

@@ -1,9 +1,6 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-
-# isort: off
-# We have to turn off isort here to resolve a conflict between isort and yapf.
 from typing import (
   Any,
   Callable,
@@ -15,7 +12,6 @@
   Tuple,
   Union,
 )
-# isort: on
 
 import chex
 import jax

prize_qualification_baselines/external_tuning/jax_nadamw_target_setting.py

@@ -1,9 +1,6 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-
-# isort: off
-# We have to turn off isort here to resolve a conflict between isort and yapf.
 from typing import (
   Any,
   Callable,
@@ -15,7 +12,6 @@
   Tuple,
   Union,
 )
-# isort: on
 
 import chex
 import jax

prize_qualification_baselines/self_tuning/jax_nadamw_full_budget.py

@@ -1,9 +1,6 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-
-# isort: off
-# We have to turn off isort here to resolve a conflict between isort and yapf.
 from typing import (
   Any,
   Callable,
@@ -15,7 +12,6 @@
   Tuple,
   Union,
 )
-# isort: on
 
 import chex
 import jax

prize_qualification_baselines/self_tuning/jax_nadamw_target_setting.py

@@ -1,9 +1,6 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-
-# isort: off
-# We have to turn off isort here to resolve a conflict between isort and yapf.
 from typing import (
   Any,
   Callable,
@@ -15,7 +12,6 @@
   Tuple,
   Union,
 )
-# isort: on
 
 import chex
 import jax

reference_algorithms/paper_baselines/nadamw/jax/submission.py

@@ -1,9 +1,6 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
 import functools
-
-# isort: off
-# We have to turn off isort here to resolve a conflict between isort and yapf.
 from typing import (
   Any,
   Callable,
@@ -15,7 +12,6 @@
   Tuple,
   Union,
 )
-# isort: on
 
 import chex
 import jax