#!/usr/bin/env python3
"""
测试 timm 导入是否正常工作
"""
import sys
print("Python version:", sys.version)

try:
    from timm.layers import to_2tuple, DropPath, trunc_normal_
    from timm.models import register_model
    print("✓ 成功导入 timm.layers.to_2tuple")
    print("✓ 成功导入 timm.layers.DropPath")
    print("✓ 成功导入 timm.layers.trunc_normal_")
    print("✓ 成功导入 timm.models.register_model")
except ImportError as e:
    print(f"✗ 导入失败: {e}")
    sys.exit(1)

try:
    from models.swiftformer import SwiftFormer_XS
    print("✓ 成功导入 SwiftFormer_XS")
except ImportError as e:
    print(f"✗ 导入 SwiftFormer_XS 失败: {e}")
    sys.exit(1)

try:
    from models.swiftformer_temporal import SwiftFormerTemporal_XS
    print("✓ 成功导入 SwiftFormerTemporal_XS")
except ImportError as e:
    print(f"✗ 导入 SwiftFormerTemporal_XS 失败: {e}")
    sys.exit(1)

print("\n✅ 所有导入测试通过！")