Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
Niccolo-Ajroldi committed Oct 3, 2024
1 parent 107c6b6 commit d4ad0eb
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Submission file for a SGD with HeavyBall momentum optimizer in Jax."""

import functools
from typing import Callable, Dict, Iterator, List, Tuple, Any
from typing import Any, Callable, Dict, Iterator, List, Tuple

from flax import jax_utils
import jax
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Submission file for a SGD with HeavyBall momentum optimizer in PyTorch."""

from typing import Callable, Dict, Iterator, List, Tuple, Any
from typing import Any, Callable, Dict, Iterator, List, Tuple

from absl import logging
import optax
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Submission file for a SGD with Nesterov momentum optimizer in Jax."""

import functools
from typing import Callable, Dict, Iterator, List, Tuple, Any
from typing import Any, Callable, Dict, Iterator, List, Tuple

from flax import jax_utils
import jax
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Submission file for a SGD with Nesterov momentum optimizer in PyTorch."""

from typing import Callable, Dict, Iterator, List, Tuple, Any
from typing import Any, Callable, Dict, Iterator, List, Tuple

from absl import logging
import optax
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Submission file for a SAM optimizer with warmup+cosine LR in PyTorch."""

from typing import Callable, Dict, Iterator, List, Tuple, Any
from typing import Any, Callable, Dict, Iterator, List, Tuple

from absl import logging
import torch
Expand Down

0 comments on commit d4ad0eb

Please sign in to comment.