# environment.yml
name: iba # The name of the environment

channels: # The conda channels to search for packages
  # - pytorch
  - conda-forge
  # - dnachun
  # - anaconda
channel_priority: strict

dependencies:
  # Packages to install with conda
  # - python=3.11.3
  #- pytorch-cuda=12.4
  # - pytorch >= 2.6
  # - numpy

  # - tensorboard
  # - omegaconf


  - wandb

  - scipy
  - pandas
  - matplotlib
  - scikit-image
  - scikit-learn
  - joblib
  - pillow
  ## NO - huggingface_hub
  - tqdm
  - nltk
  # - future
  

  # - defusedxml
  # - ipdb
  # - torchinfo



  # - timm
  # - graphviz #anaconda::graphviz
  # - dnachun::torchviz
  - pip:
    - --index-url https://download.pytorch.org/whl/cu126
    - torch
    - torchvision
    - torchaudio
    - --index-url https://pypi.org/simple
    - transformers>=4.55
    - accelerate
    
    - einops
    - jaxtyping
    - peft
    - datasets
     # - fraction
    - draccus
    - vllm