fix more isort issues

This commit is contained in:
Lincoln Stein
2023-09-15 22:22:43 -04:00
parent b9a90fbd28
commit db7fdc3555
6 changed files with 21 additions and 39 deletions

View File

@ -5,11 +5,8 @@
import argparse
import sys
from pathlib import Path
from invokeai.backend.model_manager import (
ModelProbe,
SchedulerPredictionType,
InvalidModelException,
)
from invokeai.backend.model_manager import InvalidModelException, ModelProbe, SchedulerPredictionType
def helper(model_path: Path):

View File

@ -3,14 +3,14 @@ Test the refactored model config classes.
"""
from invokeai.backend.model_manager.config import (
ModelConfigFactory,
InvalidModelConfigException,
LoRAConfig,
MainCheckpointConfig,
MainDiffusersConfig,
LoRAConfig,
TextualInversionConfig,
ModelConfigFactory,
ONNXSD1Config,
ONNXSD2Config,
TextualInversionConfig,
ValidationError,
)

View File

@ -2,18 +2,19 @@
import tempfile
import time
import requests
from requests_testadapter import TestAdapter
from requests import HTTPError
from pathlib import Path
import requests
from requests import HTTPError
from requests_testadapter import TestAdapter
import invokeai.backend.model_manager.download.queue as download_queue
from invokeai.backend.model_manager.download import (
DownloadJobBase,
DownloadJobStatus,
DownloadQueue,
DownloadJobBase,
UnknownJobIDException,
)
import invokeai.backend.model_manager.download.queue as download_queue
# Allow for at least one chunk to be fetched during the pause/unpause test.
# Otherwise pause test doesn't work because whole file contents are read

View File

@ -3,7 +3,7 @@ from pathlib import Path
import pytest
from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.backend import ModelConfigStore, BaseModelType, ModelType, SubModelType
from invokeai.backend import BaseModelType, ModelConfigStore, ModelType, SubModelType
from invokeai.backend.model_manager import ModelLoader
BASIC_MODEL_NAME = "sdxl-base-1-0"

View File

@ -2,21 +2,13 @@
Test the refactored model config classes.
"""
from hashlib import sha256
import pytest
from hashlib import sha256
from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.backend.model_manager.storage import (
ModelConfigStoreYAML,
ModelConfigStore,
UnknownModelException,
)
from invokeai.backend.model_manager.config import (
ModelType,
TextualInversionConfig,
DiffusersConfig,
VaeDiffusersConfig,
)
from invokeai.backend.model_manager.config import DiffusersConfig, ModelType, TextualInversionConfig, VaeDiffusersConfig
from invokeai.backend.model_manager.storage import ModelConfigStore, ModelConfigStoreYAML, UnknownModelException
@pytest.fixture

View File

@ -2,22 +2,14 @@
Test the refactored model config classes.
"""
import pytest
import sys
from hashlib import sha256
import pytest
from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.backend.model_manager.storage import (
ModelConfigStoreSQL,
ModelConfigStore,
UnknownModelException,
)
from invokeai.backend.model_manager.config import (
ModelType,
TextualInversionConfig,
DiffusersConfig,
VaeDiffusersConfig,
)
from invokeai.backend.model_manager.config import DiffusersConfig, ModelType, TextualInversionConfig, VaeDiffusersConfig
from invokeai.backend.model_manager.storage import ModelConfigStore, ModelConfigStoreSQL, UnknownModelException
@pytest.fixture