Skip to content

Commit e0fb2fa

Browse files
authored
Cleanup: unused imports, encoding docstring/assert, Docker diffusers pin (#409)
* fix: encoding docstring/mime + assert, Docker diffusers pin - media_to_bytes docstring claimed PNG/AVI and 'image/png'/'video/avi'; the code JPEG-encodes images and MP4-encodes video. Corrected. - video_to_bytes used 'assert video.max() <= 1.0' for input validation, which is stripped under python -O; replaced with a ValueError. - Dockerfile installed diffusers from git master on top of the pinned diffusers==0.36.0, so Docker validators ran a different diffusers than bare-metal (install.sh) and the lockfile. Dropped the git override; 0.36.0 is already installed via 'uv pip install -e .'. Janus and CLIP stay on git (not available as suitable PyPI releases). * chore: prune unused imports; declare __init__ re-exports in __all__ - ruff --fix removed 64 unused imports across gas/ and neurons/ module bodies (no behavior change; F401). - The four package __init__.py files re-export public API but had no __all__, so those imports read as 'unused'. Added explicit __all__ lists (verified against F822) — documents the public surface and lets ruff's default rules run clean. Also deleted untracked working-tree litter (cli.py copy, #VEr#, verification_ideass.txt, pr_description.md, stale emacs autosave).
1 parent 7bb6948 commit e0fb2fa

37 files changed

Lines changed: 89 additions & 69 deletions

Dockerfile

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@ RUN uv pip install setuptools && \
5252
uv pip install -e . && \
5353
uv pip install --no-build-isolation \
5454
git+https://github.com/deepseek-ai/Janus.git \
55-
git+https://github.com/huggingface/diffusers \
5655
git+https://github.com/openai/CLIP.git
5756

5857
# ---------------------------------------------------------------------------

gas/cache/db/media_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import bittensor as bt
1010

11-
from gas.cache.types import ChallengeOutcome, PromptEntry, MediaEntry
11+
from gas.cache.types import MediaEntry
1212
from gas.types import Modality, MediaType, SourceType, SOURCE_TYPE_TO_DB_NAME_FIELD
1313
from gas.cache.db.connection import ConnectionManager
1414
from gas.cache.db.prompt_store import PromptStore

gas/cache/db/prompt_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import re
44
import time
55
import uuid
6-
from typing import List, Dict, Optional, Tuple, Any
6+
from typing import List, Optional
77

88
import bittensor as bt
99

gas/cache/media_storage.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,10 @@
1-
import json
2-
import random
31
import time
42
import uuid
53
from typing import Dict, Any, Optional, List
64
from pathlib import Path
75

86
import cv2
97
import numpy as np
10-
from PIL import Image
118
from diffusers.utils import export_to_video
129

1310
from gas.cache.util.video import sample_frames

gas/cache/types.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
from dataclasses import dataclass, asdict
2-
from typing import Dict, Any, Optional, List
2+
from typing import Dict, Any, Optional
33
import numpy as np
44
import time
5-
import uuid
65

76
from gas.types import Modality, MediaType, SourceType
87

gas/cache/util/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import bittensor as bt
99
import pyarrow.parquet as pq
1010

11-
from gas.types import FileType, MediaType, Modality
11+
from gas.types import Modality
1212

1313

1414
def get_most_recent_update_time(directory: Path) -> float:

gas/cache/util/miner_stats.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import sys
1616
import time
1717
from collections import defaultdict
18-
from datetime import datetime, timezone
18+
from datetime import datetime
1919
from pathlib import Path
2020

2121

gas/datasets/__init__.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,10 @@
22
load_all_datasets,
33
get_image_datasets,
44
get_video_datasets,
5-
)
5+
)
6+
7+
__all__ = [
8+
"load_all_datasets",
9+
"get_image_datasets",
10+
"get_video_datasets",
11+
]

gas/datasets/datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"""
44

55
from typing import List
6-
from gas.types import DatasetConfig, MediaType, Modality
6+
from gas.types import DatasetConfig
77
from gasbench.dataset.config import load_benchmark_datasets_from_yaml
88

99

gas/datasets/download.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,9 @@
1-
import asyncio
21
import base64
3-
import hashlib
4-
import json
52
import random
63
import os
74
import traceback
85
import tempfile
96
import shutil
10-
from datetime import datetime
117
from io import BytesIO
128
from pathlib import Path
139
from typing import List, Optional, Dict, Any, Generator, Tuple
@@ -16,13 +12,12 @@
1612
from contextlib import closing
1713

1814
import bittensor as bt
19-
import numpy as np
2015
import pyarrow.parquet as pq
2116
from PIL import Image
2217

2318

2419
import requests
25-
from gas.types import Modality, MediaType, DatasetConfig
20+
from gas.types import Modality, DatasetConfig
2621

2722

2823
IMAGE_FILE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".bmp", ".gif", ".tiff", ".webp"}

0 commit comments

Comments
 (0)