Skip to content

Commit 6a059f9

Browse files
committed
Rename platform.py -> platform_tools.py
* avoid clash with python standard module "platform" when running `download_models.py` from ai_diffusion folder
1 parent 8c422df commit 6a059f9

File tree

11 files changed

+13
-13
lines changed

11 files changed

+13
-13
lines changed

ai_diffusion/comfy_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@
2424
from .localization import translate as _
2525
from .util import client_logger as log
2626
from .workflow import create as create_workflow
27-
from . import resources, platform, util
27+
from . import platform_tools, resources, util
2828

29-
if platform.is_macos:
29+
if platform_tools.is_macos:
3030
import os
3131

3232
if "SSL_CERT_FILE" not in os.environ:

ai_diffusion/image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pathlib import Path
99

1010
from .settings import settings, ImageFileFormat
11-
from .platform import is_linux
11+
from .platform_tools import is_linux
1212
from .util import clamp, ensure, client_logger as log
1313

1414
import struct
File renamed without changes.

ai_diffusion/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
from .resources import VerificationStatus, VerificationState
1717
from .network import download, DownloadProgress
1818
from .localization import translate as _
19-
from .platform import ZipFile, create_process, decode_pipe_bytes, determine_system_encoding
20-
from .platform import is_windows, is_macos
19+
from .platform_tools import ZipFile, create_process, decode_pipe_bytes, determine_system_encoding
20+
from .platform_tools import is_windows, is_macos
2121
from .util import client_logger as log, server_logger as server_log
2222

2323

ai_diffusion/settings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from typing import NamedTuple, Optional, Any
88
from PyQt5.QtCore import QObject, pyqtSignal
99

10-
from .platform import is_macos, is_windows
10+
from .platform_tools import is_macos, is_windows
1111
from .util import encode_json, read_json_with_comments, user_data_dir, client_logger as log
1212
from .localization import translate as _
1313

ai_diffusion/ui/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
from ..connection import ConnectionState
3434
from ..root import root
3535
from ..localization import translate as _
36-
from ..platform import get_cuda_devices
36+
from ..platform_tools import get_cuda_devices
3737
from ..util import ensure
3838
from .. import eventloop, resources, server, util
3939
from .theme import SignalBlocker, add_header, set_text_clipped, green, grey, red, yellow, highlight

ai_diffusion/ui/theme.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from ..settings import Setting
99
from ..style import Arch
1010
from ..client import Client
11-
from ..platform import is_windows
11+
from ..platform_tools import is_windows
1212
from ..util import client_logger as log
1313

1414
_palette = QGuiApplication.palette()

ai_diffusion/updates.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from . import __version__, eventloop
1212
from .network import RequestManager
1313
from .properties import ObservableProperties, Property
14-
from .platform import ZipFile
14+
from .platform_tools import ZipFile
1515
from .util import client_logger as log
1616

1717

scripts/download_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from argparse import ArgumentParser
2323

2424
sys.path.append(str(Path(__file__).parent.parent))
25-
from ai_diffusion import platform, resources
25+
from ai_diffusion import platform_tools, resources
2626
from ai_diffusion.resources import Arch, ModelRequirements, ResourceKind, ModelResource
2727
from ai_diffusion.resources import VerificationState
2828
from ai_diffusion.resources import required_models, default_checkpoints, optional_models
@@ -290,7 +290,7 @@ def verify_models(path: Path, models: set[ModelResource]):
290290

291291
backend = ModelRequirements.no_cuda
292292
if args.backend == "auto":
293-
devices = platform.get_cuda_devices()
293+
devices = platform_tools.get_cuda_devices()
294294
if any(major >= 10 for (major, minor) in devices): # Blackwell has compute capability 10.x
295295
backend = ModelRequirements.cuda_fp4
296296
elif len(devices) > 0:

tests/test_platform.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from pathlib import Path
33
import sys
44
from tempfile import TemporaryDirectory
5-
from ai_diffusion.platform import ZipFile, create_process, get_cuda_devices
5+
from ai_diffusion.platform_tools import ZipFile, create_process, get_cuda_devices
66
from ai_diffusion.util import ensure
77

88

0 commit comments

Comments
 (0)