Skip to content

Commit

Permalink
Remove rogue import, thanks IDE :/
Browse files Browse the repository at this point in the history
  • Loading branch information
rwightman committed Nov 26, 2024
1 parent bc7d224 commit 269bc08
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions tests/test_optim.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,23 +3,20 @@
These tests were adapted from PyTorch' optimizer tests.
"""
import math
import pytest
import functools
import importlib
import os
from copy import deepcopy

import pytest
import torch
from IPython.testing.decorators import skip_win32
from torch.testing._internal.common_utils import TestCase
from torch.nn import Parameter
from torch.testing._internal.common_utils import TestCase

from timm.optim import create_optimizer_v2, list_optimizers, get_optimizer_class, get_optimizer_info, OptimInfo
from timm.optim import param_groups_layer_decay, param_groups_weight_decay
from timm.scheduler import PlateauLRScheduler

import importlib
import os

torch_backend = os.environ.get('TORCH_BACKEND')
if torch_backend is not None:
importlib.import_module(torch_backend)
Expand Down

0 comments on commit 269bc08

Please sign in to comment.