# =============================================================================
# Extra dependencies for running model_runner_i18n + aipf_dm_metric eval pipeline
# Install: pip install --user -r requirements_extra.txt -i https://bytedpypi.byted.org/simple/
# =============================================================================

# ---- ByteDance internal packages (must use bytedpypi.byted.org index) --------
bytedtcc                # TCC config center SDK (required by model_runner.utils.util)
bytedeuler              # internal RPC framework, imported as `euler`
bytedtqs                # data query service (used by SummaryMetricsStage)
bytedes                 # internal Elasticsearch client (used by pairwise_comparison)

# ---- Optional ByteDance internal packages (preemptive; may not all be needed) -
# bytedabase
# bytedrh2
# bytedfirefly
# bytedfanout
# bytedmetrics
# byteddatabus

# ---- Standard PyPI packages --------------------------------------------------
scikit-learn            # cosine_similarity in similarity_dataset
aiofiles                # async file IO in dm_mm_context_dataset_v2
moviepy                 # ImageSequenceClip in dm_mm_context_dataset_v2
tensorboardX            # SummaryWriter in binary_classification_evaluator

# ---- Optional standard packages (silence import warnings) --------------------
faiss-cpu               # FaissSimilarityDataset
datasketch              # RemoveApproximateDuplicateDataset
scikit-multilearn       # stratified_dataset
interval                # stratified_dataset
langfuse                # LangfuseDataset

# ---- Packages NOT installable on bytedpypi (handled by code-level workarounds)
# bytetim               # commented out in model_runner/datasets/__init__.py
