# Atomizer - Complete Requirements
# =================================
# LLM-native FEA optimization framework with Neural Network acceleration
#
# Installation:
#   pip install -r requirements.txt
#
# For GPU support (recommended for neural training):
#   pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
#   pip install torch-geometric
#
# Note: numpy<2.0 required for pyNastran compatibility

# ============================================================================
# Core Scientific Computing
# ============================================================================
numpy>=1.24.0,<2.0.0
scipy>=1.10.0
pandas>=2.0.0
scikit-learn>=1.3.0

# ============================================================================
# Optimization Framework
# ============================================================================
optuna>=3.5.0
plotly>=5.18.0
matplotlib>=3.5.0

# ============================================================================
# FEA/Nastran Integration
# ============================================================================
pyNastran>=1.4.0,<1.5.0
h5py>=3.0.0

# ============================================================================
# Neural Network / AtomizerField
# ============================================================================
# PyTorch - Deep learning framework
torch>=2.0.0

# PyTorch Geometric - Graph neural networks
torch-geometric>=2.3.0

# TensorBoard - Training visualization
tensorboard>=2.13.0

# ============================================================================
# Web Framework (Dashboard/API)
# ============================================================================
fastapi>=0.109.0
uvicorn>=0.27.0
websockets>=12.0
pydantic>=2.5.0
python-multipart>=0.0.6
jinja2>=3.1.3

# ============================================================================
# System Utilities
# ============================================================================
psutil>=5.9.0
tqdm>=4.65.0

# ============================================================================
# Development Tools (optional)
# ============================================================================
pytest>=7.4.0
pytest-cov>=4.1.0
black>=23.12.0
ruff>=0.1.0
mypy>=1.8.0
pytest-json-report